search

AllentDan / LibtorchTutorials

This is a code repository for pytorch c++ (or libtorch) tutorial.
Apache License 2.0
742 stars 122 forks source link

how to convert trained model in libtorch to onnx? #6

Closed shining-love closed 3 years ago

shining-love commented 3 years ago

hello, thanks for your nice repo! I have a question about how to convert trained model in libtorch to onnx? could you give me some instructions?

AllentDan commented 3 years ago

This is actually what I want to know. Hope official pytorch supports this feature as soon as possible

shining-love commented 3 years ago

This is actually what I want to know. Hope official pytorch supports this feature as soon as possible

Thanks for your kind reply!I have a idea about convert trained model in libtorch(.pt) to pytorch(.pt)，is it ok？

AllentDan commented 3 years ago

I did not find a way for that before, maybe we have to dive into the PyTorch project and find a way work around

shining-love commented 3 years ago

I did not find a way for that before, maybe we have to dive into the PyTorch project and find a way work around

well，thanks for your great repo so much！I will work on convert trained model in libtorch(.pt) to pytorch(.pt).