【发布时间】:2017-09-20 03:15:42
【问题描述】:
全部。我想从一个 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 是最后一个默认节点名称)。但是,我收到一条通知“Converted 0 variables to const ops”。 (冻结命令的结果)。现在,我不知道哪个 node_name 应该作为 output_node_name?
【问题讨论】:
-
在保存之前,我打印所有节点名称。在这些节点中,“ArgMax”和“ArgMax_i”(i>=1 && i
-
我更改了代码 "return tf.add(tf.matmul(outputs[-1], weights['out']) , biases['out'], name="lstm_prediction")" .是否可以将“lstm_prediction”视为 output_node_name?
标签: tensorflow freeze lstm