View source on GitHub |
Returns CheckpointReader
for checkpoint found in ckpt_dir_or_file
.
tf.train.load_checkpoint( 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. |
© 2020 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/versions/r1.15/api_docs/python/tf/train/load_checkpoint