khadas / aml_npu_sdk

GNU General Public License v2.0
27 stars 11 forks source link

Support for pyTorch versions newer than 1.2.0 as conversion source possible? #3

Open scholz opened 2 years ago

scholz commented 2 years ago

Hi @yan-wyb ,

do you think it is possible that there will be support for pyTorch versions higher than 1.2.0? As we have currently v1.10.x and there were some changes in the format at leas tin 1.6.x. It is impossible to convert from recent pyTorch models. We are trying to use onnx as alternative but it also comes with its own bunch of challenges.

The error when converting newer models using the python convert tool you provided has also already been discussed a little in the forums: https://forum.khadas.com/t/pytorch-model-conversion/14172

Thanks for your reply!