tf.estimator.GlobalStepWaiterHook

Задерживает выполнение, пока глобальный шаг не достигнет wait_until_step.

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

tf.estimator.GlobalStepWaiterHook(
    wait_until_step
)

Этот хук задерживает выполнение, пока глобальный шаг не достигнет wait_until_step. Он используется для постепенного запуска рабочих процессов в распределённых средах. Пример использования — установка wait_until_step=int(K*log(task_id+1)) при условии, что task_id=0 — главный.

Методы

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() — нет.