Ask Your Question

Revision history [back]

click to hide/show revision 1
initial version

@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

image description

and nobody know what else:

image description

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()

    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?