ouput_node_names to convert the model into .pb format

[rohith-nagamalla]

Hi, Thank you for the implementation of the personaLab. I am trying to convert the model to .pb format so that I can use it for tensorflowjs. But it requires a parameter named as 'output_node_names'. So it could be helpful if someone could tell me what should be given as input to that parameter for this model.

[srivatsavagade]

import tensorflow as tf

Load the graph definition from the .meta file

with tf.Session() as sess: saver = tf.train.import_meta_graph('model/personlab/model.ckpt.meta')

# Retrieve the default computation graph
graph_def = tf.get_default_graph().as_graph_def()

# Save the graph definition to disk
tf.io.write_graph(graph_def, 'logdir', 'graph.pb', as_text=False)