python – 在Tensorflow中如何冻结已保存的模型
作者:互联网
这可能是一个非常基本的问题……
但是如何将检查点文件转换为单个.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知道如何正确地解析图表.
标签:python,tensorflow,tensorflow-serving,tflearn 来源: https://codeday.me/bug/20190627/1306507.html