kendryte / nncase

Open deep learning compiler stack for Kendryte AI accelerators ✨
Apache License 2.0
748 stars 181 forks source link

RuntimeError: unsupported transpose permute #672

Closed Nagisakurumi closed 1 year ago

Nagisakurumi commented 2 years ago

Describe the bug 转换onnx 模型到kmodel 的时候 在 compiler.compile()的时候报错 提示RuntimeError: unsupported transpose permute。

To Reproduce 我使用的转换脚本和我使用的onnx的模型 压缩包 convert_to_kmodel.zip 我使用的命令 : python convert_to_kmodel.py --onnx simplify.onnx --kmodel wz.kmodel --target k510 --test_size 320 320

Expected behavior 我希望能正常转换onnx模型到kmodel 并且得知这个为什么会出现这个问题,我看官方是支持的这个transpose 操作并且群里说过permute也是支持的,并且yolo的demo中也是运行出现这个同样的错误。

Origin model and code 模型和代码都在上述的压缩包中

Environment (please complete the following information):

curioyang commented 1 year ago

onnx的模型需要先进行simplify,修改后的脚本 new_convert_to_kmodel.zip 同时需要在新版本上才可以正确infer,应该是这个月底的版本(由于目前大环境的影响,可能下月初)

Nagisakurumi commented 1 year ago

所以这个修改侯的脚本 需要等新的版本1.0.9发出来之后才能用是吗

curioyang commented 1 year ago

是的

Nagisakurumi commented 1 year ago

好的