pytorch_pfn_extras.onnx.no_grad#

pytorch_pfn_extras.onnx.no_grad(fn, *args, **kwargs)#
Parameters:

  • fn (Callable[[...], Any]) –

  • args (Any) –

  • kwargs (Any) –

Return type:

Any