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