老潘的AI社区
Pytorch转ONNX新路径
模型优化
onnx
,
torchfx
,
pytorch
imoldpan
2023 年7 月 14 日 03:26
1
参考
https://github.com/pytorch/pytorch/blob/main/torch/onnx/_internal/fx/fx_onnx_interpreter.py#L317
https://github.com/pytorch/pytorch/issues/103965