XLAProfiler

class lightning.pytorch.profilers.XLAProfiler(port=9012)

기반 클래스: lightning.pytorch.profilers.profiler.Profiler

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

매개변수

port (int) – the port to start the profiler server on. An exception is raised if the provided port is invalid or busy.

start(action_name)

Defines how to start recording an action.

반환 형식

None

stop(action_name)

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

반환 형식

None