pytorch_pfn_extras.profiler.record_iterable#

pytorch_pfn_extras.profiler.record_iterable(tag, iter, divide_metric=False, use_cuda=False, enable=True, device='cpu', trace=False)#
Parameters:

  • tag (Optional[str]) –

  • iter (Iterable[_T]) –

  • divide_metric (bool) –

  • use_cuda (bool) –

  • enable (bool) –

  • device (DeviceLike) –

  • trace (Union[Tracer, bool]) –

Return type:

Iterable[_T]