About TorchDynamo

[hexisyztem]

Have you considered using torchDynamo instead of tracing for exporting the ONNX model? You can refer to the document below. https://pytorch.org/tutorials/beginner/onnx/export_simple_model_to_onnx_tutorial.html https://pytorch.org/docs/stable/torch.compiler_deepdive.html