pytorch_pfn_extras.profiler.record#

pytorch_pfn_extras.profiler.record(tag, metric=None, use_cuda=False, enable=True, device='cpu', trace=False)#
Parameters:

  • tag (Optional[str]) –

  • metric (Optional[str]) –

  • use_cuda (bool) –

  • enable (bool) –

  • device (DeviceLike) –

  • trace (Union[Tracer, bool]) –

Return type:

Generator[_ReportNotification, None, None]