XLAProfiler

class pytorch_lightning.profiler.XLAProfiler(port=9012)[source]

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)[source]

Defines how to start recording an action.

Return type

None

stop(action_name)[source]

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

Return type

None

summary()[source]

Create profiler summary in text format.

Return type

str