在 TensorFlow 中如何冻结保存的模型

Question

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

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

这些是我要转换的文件。

作为旁注，我将 tflearn 与 tensorflow 结合使用。

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