如何在TensorFlow中恢复检查点目录？

Question

我在TensorFlow tutorials使用方法训练模型（代码在这里 ）。 最后，我将模型保存在检查点目录中。 现在我想从检查点目录恢复：

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方法字符串doc：

它需要启动图表的会话。