什麼是LSTM模型的output_node_names？

全部。我想從一個LSTM模型生成凍結模型（https://github.com/roatienza/Deep-Learning-Experiments/tree/master/Experiments/Tensorflow/RNN）。在我的選擇中，我應該凍結最後一個預測節點並使用「bazel-bin/tensorflow/python/tools/freeze_graph --input_binary = true --input_graph = model_20170913/model.pb --input_checkpoint = model_20170913/model.ckpt - output_graph = model_20170913/frozen_graph.pb --output_node_names = ArgMax_52「（ArgMax_52是最後一個默認節點名稱）。但是，我收到了一條通知：「將0個變量轉換爲const操作符。」（凍結命令的結果）。現在，我不知道哪個node_name應該作爲output_node_name？什麼是LSTM模型的output_node_names？

來源

2017-09-20 Arthur Liang

保存前，我打印所有節點名稱。在這些節點中，使用「bazel-bin/tensorflow/python/tools/freeze_graph --input_binary = true --input_graph = model_20170913」ArgMax「和」ArgMax_i「（i> = 1 && i <= 52） /model.pb --input_checkpoint = model_20170913/model.ckpt --output_graph = model_20170913/frozen_graph.pb --output_node_names = ArgMax「可以生成凍結模型！ –

我改變了代碼「return tf.add（tf.matmul（outputs [-1]，weights ['out']），biasses ['out']，name =」lstm_prediction「）」。我是否可以將「lstm_prediction」視爲output_node_name？ –

如上所述，「lstm_prediction」是output_node_name。 Tensorboard幫助我瞭解圖表。

來源

2017-09-21 07:50:05

如果您能夠使用「lstm_prediction」作爲output_node_name，那麼您很有可能使用它來導出凍結的.pb文件。

來源

2017-11-03 05:49:28

什麼是LSTM模型的output_node_names？

回答

相關問題