tf.compat.v1.train.StopAtStepHook

Хук, запрашивающий остановку на указанном шаге.

Унаследован от: SessionRunHook

tf.compat.v1.train.StopAtStepHook(
    num_steps=None, last_step=None
)

Пожалуйста, ознакомьтесь с этим блокнотом о том, как мигрировать API в TF2.

Описание

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

Методы

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