how to convert .pth to onnx?

[ChweeLiong_T_Intel]

i trained a classification model using efficienet b0 and downloaded the models in weights.pth and label_schema.json. how do i convert it to .onnx?
thanks

cltee

[IntelSupport · Accepted solution]

Hi Cltee,

Thanks for reaching out.

The weights.pth and label_schema.json files are used for OpenVINO Training Extension (OTE). You may refer to Convert a PyTorch Model to ONNX and OpenVINO™ IR for the details of PyTorch conversion.

Regards,

Aznie

[IntelSupport]

Hi Cltee,

This thread will no longer be monitored since this issue has been resolved. If you need any additional information from Intel, please submit a new question. 

Regards,

Aznie