老潘的AI社区
Pytorch转ONNX新路径
模型优化
onnx
,
torchfx
,
pytorch
imoldpan
2023 年7 月 14 日 03:26
1
之前的onnx导出方式
1920×980 89.6 KB
image
1920×722 106 KB
image
1920×875 105 KB
image
1799×1159 71.3 KB
image
1920×901 124 KB
image
1731×843 62.8 KB
image
1695×417 88.9 KB
image
1695×417 88.9 KB
参考
https://onnxscript.ai/articles/2023-06-20-introducing-onnx-script/index.html
Export a PyTorch model to ONNX — PyTorch Tutorials 2.2.0+cu121 documentation
https://github.com/pytorch/pytorch/blob/main/torch/onnx/_internal/fx/fx_onnx_interpreter.py#L317
https://github.com/pytorch/pytorch/issues/103965