Pytorch model convert error

Which Khadas SBC do you use?

VIM3 Pro

Which system do you use? Android, Ubuntu, OOWOW or others?

Ubuntu 20.04

Hi,

I want to convert my own Yolov5 pytorch model to .nb and .so format. However, I am encountering the error in the image. Please help .

@ahmet_karazor
Try with my model

If it work normal, please check your model version with pytorch.

If it not work, maybe there is a problem with the conversion environment.

BTW, I suggest you convert your pytorch model to onnx, then convert onnx model with NPU convert tool.

@Frank
I tried to convert the resnet18.pt model you gave and I got the same error again.

I think the problem is in environment.

How should I build the environment? Is there a requirements.txt file you can share?

@ahmet_karazor Which version pytorch you install ?

@ahmet_karazor

Hi Ahmet,

Can you solve this problem?

@Frank ,
Environment was installed according to Yolov5 requirements.txt file (yolov5/requirements.txt at master · ultralytics/yolov5 · GitHub) .

There is Pytorch version 1.7.0.

Hi @Frank,

Do you have a solution suggestion?

@ahmet_karazor The convert tools just support 1.2.

I suggest you convert your pytorch model to onnx, then convert onnx model with NPU convert tool.