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
)

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

sync

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

sync()

Ожидание завершения любых незавершенных операций сохранения или восстановления.