Question

如果您输入以下内容，我需要以编程方式获取不依赖于目录列表和文件扩展性验证的现有检查点列表：

tf.train.get_checkpoint_state('checkpoints')

你可以看到打印这个列表，但我找不到一种方法来捕获这个打印列表，以便以后可以迭代，有没有办法做到这一点？

Answer 1

简单地说：

tf.train.get_checkpoint_state('checkpoints').all_model_checkpoint_paths

这将为您提供检查点路径列表。 CheckpointState protobuf在https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/training/checkpoint_state.proto

定义

Answer 2

checkpoint指定的python下有一个tf.Saver()文件，它包含最后n个可用的检查点文件，例如：

model_checkpoint_path: "model.ckpt-43083927"
all_model_checkpoint_paths: "model.ckpt-43083669"
all_model_checkpoint_paths: "model.ckpt-43083764"
all_model_checkpoint_paths: "model.ckpt-43083925"
all_model_checkpoint_paths: "model.ckpt-43083926"
all_model_checkpoint_paths: "model.ckpt-43083927"

Tensorflow：如何获取检查点列表

2 个答案: