pytorch_pfn_extras.onnx.pfto_exporter.export.record#
- pytorch_pfn_extras.onnx.pfto_exporter.export.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]