【发布时间】:2018-09-07 07:27:40
【问题描述】:
真正的直截了当的问题,我需要将我必须的 Tensorflow 模型转换为不支持 relu6 的格式,只支持常规 relu。我的模型采用 3 个 ckpt(检查点)文件(数据、索引和元文件)的形式。我需要能够在转换 relu 层后训练模型。有人可以指出我如何解决这个问题的正确方向。谢谢!
【问题讨论】:
标签: tensorflow activation-function
真正的直截了当的问题,我需要将我必须的 Tensorflow 模型转换为不支持 relu6 的格式,只支持常规 relu。我的模型采用 3 个 ckpt(检查点)文件(数据、索引和元文件)的形式。我需要能够在转换 relu 层后训练模型。有人可以指出我如何解决这个问题的正确方向。谢谢!
【问题讨论】:
标签: tensorflow activation-function
这需要重写张量流图。检查点中的文件之一是序列化的 tensorflow::GraphDef 协议缓冲区;在其中搜索 Relu6 并用 Relu 替换这些事件将修复图表。
但是,到目前为止,大多数 TensorFlow 训练脚本都不能通过加载训练图来工作;相反,他们在训练时重建图并在重建图的顶部加载现有检查点。如果您处于这种情况,最好将您的 python 训练代码更改为发出 tf.relu 而不是 Relu6。
【讨论】: