将训练有素的TensorFlow模型变成固定操作

时间:2018-08-27 15:51:57

标签: python tensorflow

有没有办法采用经过训练的TensorFlow模型,并将所有tf.Variables及其各自的权重(从正在运行的tf.Session内部或从检查点)转换为tf.constants值,以便可以在新的输入张量上运行模型而无需初始化或还原会话中的权重?那么我基本上可以将经过训练的模型浓缩为固定且不变的TensorFlow操作吗?

1 个答案:

答案 0 :(得分:1)

是的,有一个freeze_graph.py工具正用于此目的。

Tool Developer's Guide中对其进行了描述。您可以在Preparing models for mobile deployment部分中找到用法示例。