在 Tensorflow 中如何冻结保存的模型
In Tensorflow how to freeze saved model
这可能是一个非常基本的问题...
但是如何将检查点文件转换为单个 .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."
问题是我使用 tensorflow 而不是使用 tflearn 加载模型。
所以...而不是：
tf.train.import_meta_graph(...)
我们这样做：
model.load(...)
TFLearn 知道如何正确解析图形。
这可能是一个非常基本的问题...
但是如何将检查点文件转换为单个 .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."
问题是我使用 tensorflow 而不是使用 tflearn 加载模型。
所以...而不是：
tf.train.import_meta_graph(...)
我们这样做：
model.load(...)
TFLearn 知道如何正确解析图形。