tf.train.CheckpointManager

Управляет несколькими контрольными точками, сохраняя некоторые и удаляя ненужные.

tf.train.CheckpointManager(
    checkpoint,
    directory,
    max_to_keep,
    keep_checkpoint_every_n_hours=None,
    checkpoint_name='ckpt',
    step_counter=None,
    checkpoint_interval=None,
    init_fn=None
)

Пример использования:

import tensorflow as tf
checkpoint = tf.train.Checkpoint(optimizer=optimizer, model=model)
manager = tf.train.CheckpointManager(
    checkpoint, directory="/tmp/model", max_to_keep=5)
status = checkpoint.restore(manager.latest_checkpoint)
while True:
  # train
  manager.save()

CheckpointManager сохраняет собственное состояние при повторных инициализациях (см. __init__ документацию для получения подробностей). Только одна должна быть активна в определённом каталоге в данный момент.

Методы

restore_or_initialize

Просмотреть исходный код

restore_or_initialize()

Восстановление элементов в checkpoint из файла последней контрольной точки.

Этот метод сначала попытается восстановить из последней контрольной точки в directory. Если контрольные точки в directory не существуют, и init_fn указано, этот метод вызовет init_fn для выполнения настраиваемой инициализации. Это можно использовать для поддержки инициализации с предварительно обученных моделей.

Обратите внимание, что в отличие от tf.train.Checkpoint.restore(), этот метод не возвращает объект состояния загрузки, на котором пользователи могут выполнять утверждения (например, assert_consumed()). Таким образом, для выполнения утверждений пользователи должны напрямую использовать метод tf.train.Checkpoint.restore().

save

Просмотреть исходный код

save(
    checkpoint_number=None, check_interval=True, options=None
)

Создаёт новую контрольную точку и управляет ею.