这可能是一个非常基本的问题……
但是如何将检查点文件转换为单个.pb文件.
我的目标是使用大概C来服务模型
这些是我试图转换的文件.
作为旁注,我正在使用带有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:
也许这会对这个问题有所了解.
我得到的错误来自回归层,如果我使用:sgd.
我去拿
KeyError: "The name 'SGD' refers to an Operation not in the graph."
解决方法:
问题是我使用tensorflow而不是使用tflearn加载模型.
所以…而不是:
tf.train.import_Meta_graph(...)
我们的确是:
model.load(...)
TFLearn知道如何正确地解析图表.
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。