tf.compat.v1.train.FinalOpsHook

Хук, который вычисляет Tensors в конце сессии.

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

tf.compat.v1.train.FinalOpsHook(
    final_ops, final_ops_feed_dict=None
)

Методы

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