【问题标题】:Can I pickle a tensorflow model?我可以腌制张量流模型吗?
【发布时间】:2025-11-29 17:35:01
【问题描述】:

我能否腌制 TensorFlow 模型的所有 .meta、.data 和检查点文件?那是因为我想对我的模型进行预测,如果我部署它,文件就不能在磁盘上,对吗?我知道张量流服务,但我不太了解。我希望能够在不访问驱动器的情况下加载 tensforflow 文件。

【问题讨论】:

    标签: python tensorflow pickle


    【解决方案1】:

    不推荐使用泡菜。相反,他们创建了一种名为“SavedModel 格式”的新格式,正是为了达到这个目的。

    见:https://www.tensorflow.org/guide/saved_model

    【讨论】:

    • 我的模型没有使用 keras。它只是张量流,它保存为问题中提到的这三个文件。我已使用保护程序模型将其保存为这些文件。我想远程访问它们而不重复访问我的驱动器,这就是为什么我问泡菜是否可能。
    • 重复访问文件是什么意思?部署时将模型加载到 RAM 中一次
    最近更新 更多