如何恢復TensorFlow中的檢查點目錄？

Question

我用TensorFlow tutorials（代碼是here）的方法訓練模型。最後，我將模型保存在檢查點目錄中。現在我想從檢查點目錄恢復：

import tensorflow as tf 

def main(_): 
    saver = tf.train.Saver() 
    with tf.Session() as sess: 
     ckpt = tf.train.latest_checkpoint("/data/lstm_models") 
     saver.restore(sess, ckpt) 

if __name__ == "__main__": 
    tf.app.run() 

Howerver，我得到了錯誤：

ValueError: No variables to save 

Answer 1

它看起來像你還沒有定義您的圖表從檢查點恢復，所以建設時它會抱怨你的圖表是空的。

在嘗試恢復它之前，您可以嘗試再次構建您的圖（例如，重新定義您的變量）嗎？

從restore方法字符串文檔：

It requires a session in which the graph was launched.