tf.train.CheckpointOptions

Параметры для создания контрольной точки.

tf.train.CheckpointOptions(
    experimental_io_device=None, experimental_enable_async_checkpoint=False
)

Используется в качестве аргумента options к методам tf.train.Checkpoint.save() или tf.train.Checkpoint.restore() для настройки того, как сохраняются/восстанавливаются переменные.

Пример: выполнение операций ввода-вывода на "localhost" при сохранении контрольной точки:

step = tf.Variable(0, name="step")
checkpoint = tf.train.Checkpoint(step=step)
options = tf.train.CheckpointOptions(experimental_io_device="/job:localhost")
checkpoint.save("/tmp/ckpt", options=options)