pytorch_pfn_extras.onnx.export#

pytorch_pfn_extras.onnx.export(model, args, f, return_output=False, strip_large_tensor_data=False, large_tensor_threshold=100, chrome_tracing='', **kwargs)#

Export model into ONNX Graph.

Parameters:

  • f (IO) – A file-like object or a string file path to be written to this file.

  • return_output (bool) – If True, return output values come from the model.

  • strip_large_tensor_data (bool) – If True, this function will strip data of large tensors to reduce ONNX file size for benchmarking

  • large_tensor_threshold (int) – If number of elements of tensor is larger than this value, the tensor is stripped when strip_large_tensor_data is True

  • model (Module) –

  • args (Sequence[Any]) –

  • chrome_tracing (str) –

  • kwargs (Any) –

Return type:

Any

Warning

This function is not thread safe.