tensorflow - 如何在 Tensorflow 检查点用常规 relu 替换 relu6 操作？

Question

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

Answer 1

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

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