【发布时间】:2019-02-02 04:55:15
【问题描述】:
有没有办法采用经过训练的 TensorFlow 模型并将所有 tf.Variables 及其各自的权重(来自正在运行的 tf.Session 或来自检查点)转换为具有该值的 tf.constants,这样就可以在新的输入张量上运行模型而不初始化或恢复会话中的权重?那么我基本上可以将一个训练好的模型浓缩成一个固定不变的 TensorFlow 操作吗?
【问题讨论】:
标签: python tensorflow
有没有办法采用经过训练的 TensorFlow 模型并将所有 tf.Variables 及其各自的权重(来自正在运行的 tf.Session 或来自检查点)转换为具有该值的 tf.constants,这样就可以在新的输入张量上运行模型而不初始化或恢复会话中的权重?那么我基本上可以将一个训练好的模型浓缩成一个固定不变的 TensorFlow 操作吗?
【问题讨论】:
标签: python tensorflow
是的,有一个 freeze_graph.py 工具就是为了这个目的。
在Tool Developer's Guide 中有描述(有点)。您可以在Preparing models for mobile deployment 部分找到用法示例。
【讨论】: