【问题标题】:Tensorflow 2.0 : Variables in Saved modelTensorflow 2.0:已保存模型中的变量
【发布时间】:2019-04-07 19:43:58
【问题描述】:

saved_model API 以协议缓冲区 (pb) 格式保存的变量文件吗?如果没有,有没有办法在不使用 tensorflow API(恢复/加载)的情况下加载它

【问题讨论】:

    标签: tensorflow tensorflow2.0


    【解决方案1】:

    如果有帮助,有一个不使用 TensorFlow 操作的纯 Python API:list variablesload a single variable。对于 SavedModel,您可以将它们指向 variables/ 子目录。

    还有TensorBundle,它是C++ 中的实现。

    如果这些都没有帮助,答案可能是“否”。从理论上讲,它可以分拆成一个单独的包;如果您对此感兴趣,请随时与我们联系。

    【讨论】:

    • 嗨艾伦,感谢您的建议。我正在寻找在具有最小依赖性的嵌入式系统上的变量文件中加载张量的选项。目前最好的解决方案是冻结图,它涉及保存模型的额外对话步骤。
    【解决方案2】:

    您可以使用tf.keras.models.load_model 从saved_model 加载模型,得到的是tf.keras.Model 对象。

    【讨论】:

      【解决方案3】:

      我不确定它是否经过验证。但似乎将 List_variables 和 load_variable 指向 SavedModel 的变量子目录不起作用。我们将看到“检查点”文件错过了断言。 WA 是创建一个检查点文件,其中一行指向数据文件名。 model_checkpoint_path: "变量"

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 2020-02-07
        • 2020-02-27
        • 1970-01-01
        • 1970-01-01
        • 2019-09-18
        • 2021-01-08
        • 1970-01-01
        • 2020-03-16
        相关资源
        最近更新 更多