XLAProfiler
- class pytorch_lightning.profilers.XLAProfiler(port=9012)[source]
Bases:
pytorch_lightning.profilers.profiler.Profiler
XLA Profiler will help you debug and optimize training workload performance for your models using Cloud TPU performance tools.
- Parameters:
port (
int
) – the port to start the profiler server on. An exception is raised if the provided port is invalid or busy.