tf.train.ProfilerHook

Захватывает информацию о профилировании CPU/GPU каждые N шагов или секунд.

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

tf.train.ProfilerHook(
    save_steps=None, save_secs=None, output_dir='', show_dataflow=True,
    show_memory=False
)

Это создает файлы, называемые "timeline-<шаг>.json", которые имеют формат трассировки Chrome.

Для получения дополнительной информации см.: https://github.com/catapult-project/catapult/blob/master/tracing/README.md

Методы

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