pytorch_pfn_extras.onnx.load.load_model#
- pytorch_pfn_extras.onnx.load.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