-
Notifications
You must be signed in to change notification settings - Fork 193
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
How can I export .pth model to .onnx model? #107
Comments
@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? |
Hi @xuanhaihust, can you please share the steps for how to export .pth to onnx ? Or any one |
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). |
Has anyone solved this problem? I also faced this problem. |
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 |
I have used
![image](https://user-images.githubusercontent.com/11147205/144380313-7be71cb6-5a2f-4d11-a4bf-286da9c1871d.png)
But I can not export checkpoint to onnx model. Please help me!
The text was updated successfully, but these errors were encountered: