什么是LSTM模型的output_node_names?

时间:2017-09-20 03:15:43

标签: tensorflow freeze lstm

所有。我想从一个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?

1 个答案:

答案 0 :(得分:0)

如上所述," lstm_prediction"是output_node_name。 Tensorboard帮助我理解图表。