tf.profiler.experimental.start

Начать профилирование производительности TensorFlow.

tf.profiler.experimental.start(
    logdir, options=None
)

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

options = tf.profiler.experimental.ProfilerOptions(host_tracer_level = 3,
                                                   python_tracer_level = 1,
                                                   device_tracer_level = 1)
tf.profiler.experimental.start('logdir_path', options = options)
# Training code here
tf.profiler.experimental.stop()

Чтобы просмотреть результаты профилирования, запустите TensorBoard и укажите ему путь к logdir. Откройте свой браузер и перейдите по адресу localhost:6006/#profile для просмотра результатов профилирования.