tf.estimator.FinalOpsHook

Адаптер, который вычисляет Tensors в конце сеанса.

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

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