Import ONNX model to Pytorch

[hossein1387]

🚀 Feature

Importing ONNX models into Pytorch.

Motivation

Almost all other frameworks already support this. Importing ONNX models into Pytorch makes Pytorch much more flexible.

Pitch

In torch.onnx, a function should be created to take the ONNX model and outputs a Pytorch model.

cc @BowenBao @neginraoof

[davidtranno1]

Hi everyone,
As i know caffe2 fully supports ONNX import/export and caffe 2 is a part of PyTorch recently. However, I can not import onnx model file to Pytorch and run inference properly. How can I resolve this issue?

[olegmikul]

Agreed. I was kind a shocked when I have learnt PyTorch doesn't have that.

[rahul99]

@soumith, any plans to roll out this functionality anytime soon?

[gordinmitya]

+ to motivation
I need to convert an onnx model to torchscript (.pt) so I can run it on pytorch mobile.
How can I do this now? Any workarounds?

[jcowles]

+1, not supporting ONNX import feels like half of the ONNX implementation is missing

[kooyunmo]

Is there any ongoing PR for this issue?

[borhanMorphy]

As a heavy torch user, this feature sounds nice.

[IguanaAzul]

This feature is relevant, it should be implemented.

[LyricZhao]

Looking forward to have it in torch

[JVGD]

This feature is very relevant, my motivation:

• Load ONNX models into pytorch to compare models against same test cases

[solarflarefx]

@gordinmitya Have you found a solution to your problem? I am trying to do something similar.

[gordinmitya]

@solarflarefx no ):
pytorch mobile suddenly stoped working without any changes from my side, so I decide to just skip it until mobile version become stable