Revision history [back]

@VonChenpPlus, As you may know, there are two stages for deep learning models: training and testing. Attached .pb graph is in training phase that means it has a lot of extra nodes. In example, loss function

and nobody know what else:

To solve it, you may try to turn this model to testing phase by TensorFlow tools:

graph = 'output_graph.pb'
with tf.gfile.FastGFile(graph) as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

    inp_nodes = ['X_content']
    out_nodes = ['']
    graph_def = optimize_for_inference_lib.optimize_for_inference(graph_def, inp_nodes, out_nodes, tf.float32.as_datatype_enum)

But looking onto this graph I have no idea what the name of output node. May be you can help?