ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


TensorFlow 2 version View source on GitHub

Returns CheckpointState proto from the "checkpoint" file.

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

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

A CheckpointState if the state was available, None otherwise.

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