在Tensorflow中如何凍結已保存的模型

Question

這可能是一個非常基本的問題......

但是如何將檢查點文件轉換為單個.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."

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知道如何正確地解析圖表。