tf.train.load_checkpoint
Возвращает CheckpointReader для контрольной точки, найденной в ckpt_dir_or_file.
tf.train.load_checkpoint(
ckpt_dir_or_file
)
Использование в блокнотах
| Используется в руководстве |
| |
Если ckpt_dir_or_file разрешает папку с несколькими контрольными точками, возвращается ридер для последней контрольной точки.
Пример использования:
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
| Аргументы |
|---|
ckpt_dir_or_file | Директория с файлами контрольных точек или путь к файлу контрольной точки. |
| Возвращает |
|---|
Объект CheckpointReader. |
| Возможные исключения |
|---|
ValueError | Если ckpt_dir_or_file разрешает папку без контрольных точек. |
