tf.train.load_checkpoint

TensorFlow 2 version View source on GitHub

Returns CheckpointReader for checkpoint found in ckpt_dir_or_file.

If ckpt_dir_or_file resolves to a directory with multiple checkpoints, reader for the latest checkpoint is returned.

ckpt_dir_or_file Directory with checkpoints file or path to checkpoint file.

CheckpointReader object.

ValueError If ckpt_dir_or_file resolves to a directory with no checkpoints.