【问题标题】:What's the LSTM model's output_node_names?LSTM 模型的 output_node_names 是什么?
【发布时间】: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


【解决方案1】:

如上所述,“lstm_prediction”是 output_node_name。 Tensorboard 对我理解图表有很大帮助。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2019-06-11
    • 1970-01-01
    • 2020-08-04
    • 2020-11-26
    • 1970-01-01
    • 2020-05-28
    • 2020-05-25
    • 1970-01-01
    相关资源
    最近更新 更多