tf.compat.v1.train.SummarySaverHook

Сохраняет сводки каждые N шагов.

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

tf.compat.v1.train.SummarySaverHook(
    save_steps=None,
    save_secs=None,
    output_dir=None,
    summary_writer=None,
    scaffold=None,
    summary_op=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, которые вы возвращаете, также могут содержать значения feed, которые нужно добавить в вызов run().

Аргумент run_context - объект SessionRunContext, который предоставляет информацию о предстоящем вызове run(): первоначально запрошенные операции/тензоры, сессия TensorFlow.

На этом этапе граф финализирован, и операции добавить нельзя.

begin

Просмотреть исходный код

begin()

Вызывается один раз перед использованием сессии.

При вызове используется текущий граф, который будет запущен в сессии. Хук может изменять граф, добавляя новые операции. После вызова begin() граф будет финализирован, и другие обратные вызовы больше не смогут изменять граф. Второй вызов begin() на одном и том же графе не должен изменять граф.

end

Просмотреть исходный код

end(
    session=None
)

Вызывается в конце сессии.

Аргумент session может быть использован, если хук хочет выполнить заключительные операции, такие как сохранение последнего контрольного пункта.

Если session.run() вызывает исключение, отличное от OutOfRangeError или StopIteration, то end() не вызывается. Обратите внимание на разницу в поведении end() и after_run(), когда session.run() вызывает OutOfRangeError или StopIteration. В этом случае end() вызывается, но after_run() не вызывается.