TensorFlow 2 version | 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. |