How to export pt model

[lizhiwen19900709]

How to customize labels export pt model files ？

[taofuyu]

this line

[lizhiwen19900709]

This is the export onnx model, I want to export pt model @taofuyu

[wondervictor]

@lizhiwen19900709, does pt denote the pytorch model?

[lizhiwen19900709]

@lizhiwen19900709, does pt denote the pytorch model?

yes，.pt model files

[wondervictor]

exactly, the pt and pth models are the same.

[lizhiwen19900709]

exactly, the pt and pth models are the same.

Eventually I'm going to switch to the coreml model，Before the successful test pt to corem, so want to transfer pt

[wondervictor]

Hi @lizhiwen19900709, according to my comprehension based on CoreML Docs, you need the torch model with checkpoints (pt or pth) and convert the model to a jit model and then to a CoreML model.

[lizhiwen19900709]

Can you help me write a JIT script? I'm not familiar with the parameters. I've been tossing around for a long time. I've been failing.

Hi @lizhiwen19900709, according to my comprehension based on CoreML Docs, you need the torch model with checkpoints (pt or pth) and convert the model to a jit model and then to a CoreML model.