ENOT-AutoDL / onnx2torch

Convert ONNX models to PyTorch.
Apache License 2.0
610 stars 71 forks source link

Can it provide network code from onnx ? #219

Closed lix19937 closed 3 months ago

lix19937 commented 5 months ago

Can this repo provide network code from onnx, not a pth file ?

senysenyseny16 commented 3 months ago

hi @lix19937,

onnx2torch can generate a GraphModule only. For more info, see https://pytorch.org/docs/stable/fx.html.

lix19937 commented 3 months ago

Thanks!