tf.train.get_checkpoint_state

View source on GitHub

Returns CheckpointState proto from the "checkpoint" file.

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

If the "checkpoint" file contains a valid CheckpointState proto, returns it.

Args:

• checkpoint_dir: The directory of checkpoints.
• latest_filename: Optional name of the checkpoint file. Default to 'checkpoint'.

Returns:

A CheckpointState if the state was available, None otherwise.

Raises:

• ValueError: if the checkpoint read doesn't have model_checkpoint_path set.