How can I export .pth model to .onnx model?

[thanhtung0406]

I have used

But I can not export checkpoint to onnx model. Please help me!

[chriscpy]

@thanhtung0406 Have you solved this problem?I also encountered the same problem

[chriscpy]

@wenwenyu

[xuanhaihust]

@thanhtung0406 Have you solved this problem?I also encountered the same problem

We solved it, but in a horrible way and you should not try.

[AtulKumar4]

@thanhtung0406 Have you solved this problem?I also encountered the same problem

We solved it, but in a horrible way and you should not try.

@xuanhaihust Could you share the steps if possible?
Thanks

[RahulJha11]

Hi @xuanhaihust, can you please share the steps for how to export .pth to onnx ? Or any one
solve this issue.

[xuanhaihust]

Hi @RahulJha11, @AtulKumar4 Sorry for the late answer. We could not do it right away, but after break the model to smaller components, we did it and end up with 8 parts (7 .onnx files and 1 torch script).
If you want more details, contact me via Telegram @xuanhaihust or Linkedin @xuanhaihust also.

[thanhsang298]

Has anyone solved this problem? I also faced this problem.
Thank you

[cqray1990]

Hi @RahulJha11, @AtulKumar4 Sorry for the late answer. We could not do it right away, but after break the model to smaller components, we did it and end up with 8 parts (7 .onnx files and 1 torch script). If you want more details, contact me via Telegram @xuanhaihust or Linkedin @xuanhaihust also.

the model you convert to ,can infer by dynamic_axis ? torch.onnx.export(dynamic_axis=dynamic_axis)

@xuanhaihust whick version of pytorch and onnx you use when convert to onnx