W3cubDocs

/TensorFlow Python

tf.train.get_checkpoint_state

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

Defined in tensorflow/python/training/saver.py.

See the guide: Variables > Saving and Restoring Variables

Returns CheckpointState proto from the "checkpoint" file.

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.

© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/get_checkpoint_state