【问题标题】:Modifications in evaluate.py for other architectures in tensorflow poets对 tensorflow 诗人中其他架构的评估.py 中的修改
【发布时间】:2018-06-19 06:15:34
【问题描述】:

tensorflow-for-poets-2/scripts/evaluate.py 有一个用于衡量模型性能的脚本。该脚本最初是为 mobilenet 模型编写的,但可以针对其他架构进行相应修改。我应该在41 行写什么

logits = graph.get_tensor_by_name("final_training_ops/Wx_plus_b/add:0"). 

我得到了`KeyError:

名称 final_training_ops/Wx_plus_b/add:0' 指的是一个张量,它 不存在。操作,'final_training_ops/Wx_plus_b/add', 图中不存在。

【问题讨论】:

    标签: python tensorflow convolutional-neural-network


    【解决方案1】:

    这是我自己想出来的。给定一个 .pb 文件,我们可以使用 code 在冻结的 .pb TensorFlow 图中查看操作。因此,我们需要根据我们的 .pb 文件将 41 行替换为最后一个操作。

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-08-07
      相关资源
      最近更新 更多