XLAProfiler¶
- class lightning.pytorch.profilers.XLAProfiler(port=9012)[source]¶
Bases:
Profiler
XLA Profiler will help you debug and optimize training workload performance for your models using Cloud TPU performance tools.
Bases: Profiler
XLA Profiler will help you debug and optimize training workload performance for your models using Cloud TPU performance tools.