search

rockchip-linux / rknn-toolkit

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

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

Open pumpkin639 opened 10 months ago

pumpkin639 commented 10 months ago

error

broljx commented 9 months ago

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

eRaul commented 8 months ago

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