tf.profiler.experimental.Trace

Контекстный менеджер, который генерирует событие отслеживания в профилировщике.

tf.profiler.experimental.Trace(
    name, **kwargs
)

Событие отслеживания начнётся при входе в контекст и остановится, сохранив результат в профилировщике, при выходе из контекста. Откройте вкладку Профилирование TensorBoard и выберите просмотрщик событий отслеживания, чтобы просмотреть событие отслеживания во временной шкале.

События отслеживания создаются только при включенном профилировщике. Дополнительную информацию о том, как использовать профилировщик, можно найти на странице https://tensorflow.org/guide/profiler

Пример использования:

tf.profiler.experimental.start('logdir')
for step in range(num_steps):
  # Creates a trace event for each training step with the step number.
  with tf.profiler.experimental.Trace("Train", step_num=step):
    train_fn()
tf.profiler.experimental.stop()

tf.profiler.experimental.start('logdir')
for step in range(num_steps):
# Creates a trace event for each training step with the
# step number.
with tf.profiler.experimental.Trace("Train", step_num=step):
train_fn()
tf.profiler.experimental.stop()

Методы

set_metadata

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

set_metadata(
    **kwargs
)

Устанавливает метаданные в данном событии отслеживания.

Этот метод позволяет устанавливать метаданные в событии отслеживания после его создания.

Пример использования:

def call(function):
  with tf.profiler.experimental.Trace("call",
       function_name=function.name) as tm:
    binary, in_cache = jit_compile(function)
    tm.set_metadata(in_cache=in_cache)
    execute(binary)

В этом примере мы хотим отследить, сколько времени уходит на вызов функции, включая компиляцию и выполнение. Компиляция может заключаться как в получении кэшированной копии двоичного файла, так и в фактическом генерации двоичного файла, что указывается булевым значением «in_cache», возвращаемым jit_compile(). Нам необходимо использовать set_metadata() для передачи in_cache, так как значение in_cache нам было неизвестно при создании отслеживания (и мы не можем создать отслеживание после jit_compile(), так как хотим измерить весь период выполнения вызова).

__enter__

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

__enter__()

__exit__

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

__exit__(
    exc_type, exc_val, exc_tb
)