XLAProfiler

class pytorch_lightning.profiler.XLAProfiler(port=9012)

Bases: pytorch_lightning.profiler.base.BaseProfiler

This Profiler will help you debug and optimize training workload performance for your models using Cloud TPU performance tools.

start(action_name)

Defines how to start recording an action.

Return type

None

stop(action_name)

Defines how to record the duration once an action is complete.

Return type

None

summary()

Create profiler summary in text format.

Return type

str