如何在 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()

但是，我得到了错误：

ValueError: No variables to save

Answer 1

您似乎尚未定义要从检查点恢复的图表，因此在构建保护程序时它会抱怨您的图表为空。

您能否在尝试恢复图表之前再次尝试构建图表（例如重新定义变量）？

来自restore方法字符串文档：

它需要一个启动图表的会话。