Convert ONNX to PyTorch and Train
nomaad42 opened this issue · comments
nomaad commented
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'>?
LLC commented
Can you share how to convert onnx model to pytorch model?
nomaad commented
Can you share how to convert onnx model to pytorch model?
Check out this link