Tensorflow - .pbとしてチェックポイント・ファイルを保存し、ない、出力ノード名

Question

と私は、次のファイルを持っている：

model.ckpt-2400.data-00000-of-00001 
model.ckpt-2400.index 
model.ckpt-2400.meta 

そして、私は次の関数で.pbの形式で保存したいと思います：

​​

問題は、私はtf.get_default_graph().as_graph_def().nodeを使用する場合、それは[]を返すということです。空の配列。これに使用できる出力ノード名はありません。

他にどのようにそれらを.pbとして保存できますか？私はちょうどtf.python.tools.freeze_graph.freeze_graph()関数を参照すべきですか？

Answer 1

出力ノードの名前を指定するだけです。結果を確認するためにログに記録するノードとして指定されたコードの別の部分に出力ノードの名前を指定します。

predictions = { 
     # Generate predictions (for PREDICT and EVAL mode) 
     "classes": tf.argmax(input=logits, axis=1), 
     # Add `softmax_tensor` to the graph. It is used for PREDICT and by the 
     # `logging_hook`. 
     "probabilities": tf.nn.softmax(logits, name="softmax_tensor") #This one 
    } 

私の場合はsoftmax_tensorです。