tf.train.CheckpointManager

Удаляет старые контрольные точки.

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

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

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

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

Методы

save

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

save(
    checkpoint_number=None
)

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