tf.train.get_checkpoint_state用法

tf.train.get_checkpoint_state：从“检查点”文件返回CheckpointState原型。其中有model_checkpoint_path和all_model_checkpoint_paths两个属性。其中model_checkpoint_path保存了最新的tensorflow模型文件的路径+文件名，是个字符串。all_model_checkpoint_paths则有未被删除的所有tensorflow模型文件的路径+文件名，是个列表。

checkpoint_dir = "./train_model/model/"
ckpt = tf.train.get_checkpoint_state(checkpoint_dir)
print(ckpt)
print(ckpt.model_checkpoint_path)
print(ckpt.all_model_checkpoint_paths)


model_checkpoint_path: "./train_model/model/test.ckpt-17000"
all_model_checkpoint_paths: "./train_model/model/test.ckpt-16200"
all_model_checkpoint_paths: "./train_model/model/test.ckpt-16400"
all_model_checkpoint_paths: "./train_model/model/test.ckpt-16600"
all_model_checkpoint_paths: "./train_model/model/test.ckpt-16800"
all_model_checkpoint_paths: "./train_model/model/test.ckpt-17000"


./train_model/model/test.ckpt-17000
['./train_model/model/test.ckpt-16200', 
 './train_model/model/test.ckpt-16400', 
 './train_model/model/test.ckpt-16600', 
 './train_model/model/test.ckpt-16800', 
 './train_model/model/test.ckpt-17000']

tf.train.get_checkpoint_state(
    checkpoint_dir, latest_filename=None
)

如果“检查点”文件包含有效的CheckpointState proto，则将其返回。

Args
checkpoint_dir	检查点checkpoints的目录。
latest_filename	检查点checkpoint文件的可选名称。默认为“检查点checkpoint”。

Returns
一个CheckpointState（如果状态可用），否则为None。

Raises
ValueError	如果读取的检查点未设置model_checkpoint_path。