Conversion to ONNX

[dakl]

Do you happen to have examples on how to convert the PE models to ONNX?

[shuminghu]

PE is getting ported to TIMM right now. Hopefully this could help export to ONNX.

[shuminghu]

PE is now available in timm.
huggingface/pytorch-image-models#2487
It can be exported with onnx_export.py in the timm repo root dir.

[irfanrah]

@dakl I’ve just finished converting the model to ONNX format. You can check it out here:
🔗 https://github.com/irfanrah/PE_onnx_export