pytorch_pfn_extras.onnx.load_model#

pytorch_pfn_extras.onnx.load_model(f, format=None, load_external_data=True)#

Load model from ONNX file.

This is a wrapper to onnx.load_model that automatically falls back to load_external_data=False when tensors are stripped.

Parameters:

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

  • format (Optional[Any]) – A reserved arg

  • load_external_data (bool) – If True and the external data under the same directory of the model, load the external data

Return type:

ModelProto