Export to ONNX

[sctrueew]

Hi,
Is it possible to run it with ONNX?

I really appreciate any help you can provide.

[EasternJournalist]

Hi. I have tried a few strategies with pytorch built-in ONNX tools but failed unfortunately. I am not familiar with ONNX and don't know how to fix it. I am sorry that I can't help : ( Maybe you can have a try with some other tools. Please let me know if you have any solution.

[iariav]

Hi @sctrueew and @EasternJournalist ,
I have successfully imported the model to ONNX and validated that the outputs of the original and converted models are identical.
If you want, i can submit a PR of the script.
However, it did also require me to make some minor changes (mainly naming of functions) in the moge_model.py file, so we'll need to think how to bypass that neatly.