pytorch_pfn_extras.profiler.TimeSummary.report

TimeSummary.report(tag, use_cuda=False)

Context manager to automatically report execution times.

The start and completion times are obtained automatically, the user only needs to provide a tag to identify the value in the summary values.

Parameters

• tag (str) – A name to identify the section of code being profiled.

• use_cuda (bool) – Indicates if GPU time should also be profiled.

Return type

Generator[pytorch_pfn_extras.profiler._time_summary._ReportNotification, None, None]

pytorch_pfn_extras.logging.get_logger pytorch_pfn_extras.nn.parallel.DistributedDataParallel