如何在Tensorflow检查点中使用常规relu替换relu6操作?

时间:2018-03-28 14:37:45

标签: tensorflow activation-function

直截了当的问题,我需要将我所拥有的Tensorflow模型转换为不支持relu6的格式,只需定期relu。我的模型是3 ckpt(检查点)文件(数据,索引和元文件)的形式。我需要能够在转换relu层后训练模型。有人能指出我正确的方向来解决这个问题。谢谢!

1 个答案:

答案 0 :(得分:0)

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

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