Convert ONNX to PyTorch and Train

Open nomaad42 opened this issue 3 years ago • 3 comments

I converted your ONNX model to PyTorch, and now I am trying to Train it. When I converted to PyTorch, it only accepts this tensor size: [1, 3, 224, 224], and does not accept other batch sizes.

How can I train the converted model with various batch sizes (4, 8, 16, etc)?

In addition, is it possible to train this model, as its type is <class 'torch.fx.graph_module.GraphModule.new..GraphModuleImpl'>?

Nov 18 '22 10:11 nomaad42

Can you share how to convert onnx model to pytorch model?

Feb 27 '23 15:02 LLC

Can you share how to convert onnx model to pytorch model?

Check out this link

https://github.com/ENOT-AutoDL/onnx2torch/issues/113#event-8318350020

Feb 28 '23 06:02 nomaad42