W3cubDocs

/TensorFlow Python

tf.train.load_checkpoint

tf.train.load_checkpoint(ckpt_dir_or_file)

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

Returns CheckpointReader for checkpoint found in ckpt_dir_or_file.

If ckpt_dir_or_file resolves to a directory with multiple checkpoints, reader for the latest checkpoint is returned.

Args:

  • ckpt_dir_or_file: Directory with checkpoints file or path to checkpoint file.

Returns:

CheckpointReader object.

Raises:

  • ValueError: If ckpt_dir_or_file resolves to a directory with no checkpoints.

© 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/load_checkpoint