thuml/iTransformer

Export model to ONNX

Closed this issue · 2 comments

Temppus commented

Hi, great open source implementation.
I am very new to ML ecosystem and I am trying to export trained itransformer model to ONNX, but currently failing to do so.
Do you think it would be possible to include sample or guide me how to do it ?
Thank you.

Temppus commented

Nevermind it seems like I managed to export it successfully

  • 👍1
WenWeiTHU commented

That sounds good! We will appreciate it greatly if you 'd like to get involved in this repo and pr this part.