tf.train.get_checkpoint_state总是没有

Question

我使用tf.train.get_checkpoint_state来检查是否有一个有效的检查点文件来恢复Tensorflow保存的模型。

ckpt = tf.train.get_checkpoint_state(os.path.dirname('fi/saves'))
if ckpt and ckpt.model_checkpoint_path:
   saver.restore(sess, ckpt.model_checkpoint_path)  

我的问题是，克普特从来都不是。以下是我的保存目录的内容：

• internals.pkl
• stats.json
• variables.ckpt.data-00000-of-00001
• variables.ckpt.index

请注意，直接恢复的调用运行良好。

编辑：

我尝试过这样使用latest_filename参数：

tf.train.get_checkpoint_state(os.path.dirname(checkpoint_dir), latest_filename='variables.ckpt.index')  

我还没拿到呢。

Answer 1

我发现tf.train.get_checkpoint_state在寻找检查点文件。

我没有这个文件，因为我是在保存时这样做的：

saver.save(sess, variables_file_path, write_meta_graph=False, write_state=False)

而不是：

saver.save(sess, variables_file_path, write_meta_graph=False, write_state=True)  # write state is true by default.

现在起作用了！