tf.estimator.CheckpointSaverHook

Сохраняет контрольные точки каждые N шагов или секунд.

Наследуется от: SessionRunHook

tf.estimator.CheckpointSaverHook(
    checkpoint_dir,
    save_secs=None,
    save_steps=None,
    saver=None,
    checkpoint_basename='model.ckpt',
    scaffold=None,
    listeners=None,
    save_graph_def=True
)

Методы

after_create_session

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

after_create_session(
    session, coord
)

Вызывается при создании новой TensorFlow сессии.

Это вызывается для сигнализации хукам о создании новой сессии. Это имеет два основных отличия от ситуации, в которой вызывается begin:

• Когда это вызывается, граф завершен, и к графу больше нельзя добавлять операции.
• Этот метод также будет вызван в результате восстановления обернутой сессии, а не только в начале всей сессии.

after_run

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

after_run(
    run_context, run_values
)

Вызывается после каждого вызова run().

Аргумент run_values содержит результаты запрошенных операций/тензоров вызовом before_run().

Аргумент run_context — тот же, что и отправленный в вызов before_run. Можно вызвать run_context.request_stop() для остановки итерации.

Если session.run() вызывает какие-либо исключения, то after_run() не вызывается.

before_run

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

before_run(
    run_context
)

Вызывается перед каждым вызовом run().

Можно вернуть из этого вызова объект SessionRunArgs, указывающий операции или тензоры для добавления к предстоящему вызову run(). Эти операции/тензоры будут выполнены вместе с операциями/тензорами, изначально переданными в исходный вызов run(). Аргументы вызова run, которые вы возвращаете, также могут содержать данные для добавления к вызову run().

Аргумент run_context — SessionRunContext, который предоставляет информацию о предстоящем вызове run(): изначально запрошенные операции/тензоры, TensorFlow сессия.

На этом этапе граф завершён, и вы не можете добавлять операции.

begin

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

begin()

Вызывается один раз перед использованием сессии.

При вызове, используется стандартная графа, которая будет запущена в сессии. Хук может изменить графу, добавив новые операции. После вызова begin() графа будет завершена, и другие обработчики больше не смогут изменить графу. Второй вызов begin() для той же графы не должен изменять графу.

end

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

end(
    session
)

Вызывается в конце сессии.

Аргумент session может быть использован в случае, если обработчик хочет выполнить конечные операции, такие как сохранение последнего контрольного пункта.

Если session.run() вызывает исключение, отличное от OutOfRangeError или StopIteration, то end() не вызывается. Обратите внимание на разницу в поведении end() и after_run() при том, что session.run() вызывает исключения OutOfRangeError или StopIteration. В этом случае end() вызывается, но after_run() не вызывается.