微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

python – 在Tensorflow中如何冻结已保存的模型

这可能是一个非常基本的问题……

但是如何将检查点文件转换为单个.pb文件.
我的目标是使用大概C来服务模型

这些是我试图转换的文件.

Checkpoints

作为旁注,我正在使用带有tensorflow的tflearn.

编辑1:
我发现了一篇解释如何执行此操作的文章https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc

问题是我遇到了以下错误

KeyError: "The name 'Adam' refers to an Operation not in the graph."

我该如何解决

编辑2:
也许这会对这个问题有所了解.

Regression Layer

我得到的错误来自回归层,如果我使用:sgd.
我去拿

KeyError: "The name 'SGD' refers to an Operation not in the graph."

解决方法:

https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc上的教程工作得很好.

问题是我使用tensorflow而不是使用tflearn加载模型.

所以…而不是:

tf.train.import_Meta_graph(...)

我们的确是:

model.load(...)

TFLearn知道如何正确地解析图表.

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。

相关推荐