SimpleProfiler

class lightning.pytorch.profilers.SimpleProfiler(dirpath=None, filename=None, extended=True)[source]

Bases: Profiler

This profiler simply records the duration of actions (in seconds) and reports the mean duration of each action and the total time spent over the entire training run.

Parameters:
  • dirpath (Union[str, Path, None]) – Directory path for the filename. If dirpath is None but filename is present, the trainer.log_dir (from TensorBoardLogger) will be used.

  • filename (Optional[str]) – If present, filename where the profiler results will be saved instead of printing to stdout. The .txt extension will be used automatically.

  • extended (bool) – If True, adds extra columns representing number of calls and percentage of total time spent on respective action.

Raises:

ValueError – If you attempt to start an action which has already started, or if you attempt to stop recording an action which was never started.

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