【问题标题】:How to replace relu6 operations with regular relu in Tensorflow checkpoint?如何在 Tensorflow 检查点用常规 relu 替换 relu6 操作?
【发布时间】:2018-09-07 07:27:40
【问题描述】:

真正的直截了当的问题,我需要将我必须的 Tensorflow 模型转换为不支持 relu6 的格式,只支持常规 relu。我的模型采用 3 个 ckpt(检查点)文件(数据、索引和元文件)的形式。我需要能够在转换 relu 层后训练模型。有人可以指出我如何解决这个问题的正确方向。谢谢!

【问题讨论】:

    标签: tensorflow activation-function


    【解决方案1】:

    这需要重写张量流图。检查点中的文件之一是序列化的 tensorflow::GraphDef 协议缓冲区;在其中搜索 Relu6 并用 Relu 替换这些事件将修复图表。

    但是,到目前为止,大多数 TensorFlow 训练脚本都不能通过加载训练图来工作;相反,他们在训练时重建图并在重建图的顶部加载现有检查点。如果您处于这种情况,最好将您的 python 训练代码更改为发出 tf.relu 而不是 Relu6。

    【讨论】:

      猜你喜欢
      • 2018-11-23
      • 2022-12-09
      • 2017-03-30
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2016-12-03
      • 1970-01-01
      • 2011-06-25
      相关资源
      最近更新 更多