tf.compat.v1.train.SessionRunHook

Перехватчик для расширения вызовов MonitoredSession.run().

Используется в блокнотах

Методы

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_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() не вызывается.