rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
771 stars 175 forks source link

分类模型onnx可以推理,转rknn报错,请问是否与使用全连接层算子有关,还是其他什么原因,报错信息如下,请大神指点 #436

Open pumpkin639 opened 5 months ago

pumpkin639 commented 5 months ago

error

broljx commented 4 months ago

我也遇到这种onnx模型推理正常,但是也是类似你的报错的情况,你这边解决了吗?

eRaul commented 4 months ago

跟算子无关,可能是reshape的问题。可以在config接口中把batch_size设成1来尝试规避这个问题。