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)  

我的问题是ckpt始终为None。这是我的保存目录的内容：

• 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.

现在可以了！