When will PyTorch support importing from ONNX?

1

Hi everyone,

I am currently working on a project for which PyTorch would be the ideal framework. Unfortunately, it does not yet support importing models from ONNX which is necessary for me. Does someone know when a version of PyTorch will be published, that supports importing .onnx models. On the ONNX github page it says that this feature will be “coming soon”.

Thanks in advance

2

Yes, two ways ONNX (import-export) is available for other frameworks. Would be nice to know if/when it will be available for PyTorch, making that framework more competitive versus others.

3

I’m not sure if this was a recent addition but the docs here

mention you can do something like

import onnx

# Load the ONNX model
model = onnx.load("alexnet.onnx")

# Check that the IR is well formed
onnx.checker.check_model(model)

# Print a human readable representation of the graph
onnx.helper.printable_graph(model.graph)