python - 查找Tensorflow中的所有检查点路径

Question

到目前为止，我在 Tensorflow 中保存和加载检查点仅用于加载最后一个检查点。通常我使用的代码是这样的:

ckpt = tf.train.get_checkpoint_state(load_dir)
if ckpt and ckpt.model_checkpoint_path:
    saver.restore(session, ckpt.model_checkpoint_path)
else:
    tf.gfile.DeleteRecursively(load_dir)
    tf.gfile.MakeDirs(load_dir)

但是，在我最新的实验中，我每 1000 次迭代保存一个检查点，并且我想在所有检查点上运行评估脚本，例如显示不同验证指标的进展情况。有没有简单的方法可以获取 Tensorflow 中的所有检查点，或者我只需要使用 os 相应地循环所有名称？

Answer 1

代码片段中的 ckpt 对象是 CheckpointState Protocol Buffer 。您可以使用以下方法迭代所有模型路径，而不是访问最新的模型路径 (ckpt.model_checkpoint_path):

for model_path in ckpt.all_model_checkpoint_paths:
    saver.restore(session, model_path)
    # Do the evaluation using the restored model