tf.train.load_checkpoint | TensorFlow v2.16.1 (original) (raw)

tf.train.load_checkpoint

Stay organized with collections Save and categorize content based on your preferences.

Returns CheckpointReader for checkpoint found in ckpt_dir_or_file.

View aliases

Compat aliases for migration

tf.compat.v1.train.load_checkpoint

tf.train.load_checkpoint(
    ckpt_dir_or_file
)

Used in the notebooks

Used in the guide
Migrating model checkpoints Training checkpoints Migrate the fault tolerance mechanism

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

Example usage:

import tensorflow as tf
a = tf.Variable(1.0)
b = tf.Variable(2.0)
ckpt = tf.train.Checkpoint(var_list={'a': a, 'b': b})
ckpt_path = ckpt.save('tmp-ckpt')
reader= tf.train.load_checkpoint(ckpt_path)
print(reader.get_tensor('var_list/a/.ATTRIBUTES/VARIABLE_VALUE'))  # 1.0

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.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.