编程语言
首页 > 编程语言> > python – 在Tensorflow中如何冻结已保存的模型

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知道如何正确地解析图表.

标签:python,tensorflow,tensorflow-serving,tflearn
来源: https://codeday.me/bug/20190627/1306507.html