tf.train.get_checkpoint_state

Returns CheckpointState proto from the "checkpoint" file.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.train.get_checkpoint_state, `tf.compat.v2.train.get_checkpoint_state`

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.