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

Question

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

但是如何将检查点文件转换为单个.pb文件。 我的目标是使用可能的C ++来提供模型

这些是我要转换的文件。

作为旁注，我正在使用带有张量流的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."

Answer 1

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