tf.train.CheckpointOptions

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

tf.train.CheckpointOptions(
    experimental_io_device=None
)

Используется в качестве аргумента _options конструктора tf.Checkpoint, чтобы настроить сохранение переменных.

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

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