【问题标题】:Turn trained TensorFlow model into fixed operation将训练好的 TensorFlow 模型转为固定操作
【发布时间】:2019-02-02 04:55:15
【问题描述】:

有没有办法采用经过训练的 TensorFlow 模型并将所有 tf.Variables 及其各自的权重(来自正在运行的 tf.Session 或来自检查点)转换为具有该值的 tf.constants,这样就可以在新的输入张量上运行模型而不初始化或恢复会话中的权重?那么我基本上可以将一个训练好的模型浓缩成一个固定不变的 TensorFlow 操作吗?

【问题讨论】:

    标签: python tensorflow


    【解决方案1】:

    是的,有一个 freeze_graph.py 工具就是为了这个目的。

    Tool Developer's Guide 中有描述(有点)。您可以在Preparing models for mobile deployment 部分找到用法示例。

    【讨论】:

    • 谢谢,这正是我需要的!
    猜你喜欢
    • 1970-01-01
    • 2017-11-17
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2021-09-15
    • 1970-01-01
    • 2017-11-11
    • 1970-01-01
    相关资源
    最近更新 更多