search

rockchip-linux / rknn-toolkit

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

Cannot reshape a tensor with 25600000 elements to shape [100,80,20,20] 求大佬看看这个怎么解决呀 #425

Closed 17656178609 closed 3 weeks ago

17656178609 commented 6 months ago

log_feedback_to_the_rknn_toolkit_dev_team.log ValueError: Cannot reshape a tensor with 25600000 elements to shape [100,80,20,20] (3200000 elements) for '{{node Reshape_/model.10/Reshape294/Reshape/model.10/Reshape_294}} = Reshape[T=DT_FLOAT, Tshape=DTINT32](Reshape/model.10/Reshape_294_RKNN_mark_perm918/Reshape/model.10/Reshape_294_RKNN_mark_perm918, Reshape/model.10/Reshape294/Reshape/model.10/Reshape_294/shape)' with input shapes: [100,640,20,20], [4] and with input tensors computed as partial shapes: input[1] = [100,80,20,20].

eRaul commented 6 months ago

在config接口中把batch_size设成1试试。

KopiSoftware commented 3 months ago

在config接口中把batch_size设成1试试。

Thank you VERY much. It works in my case when I try to convert rtmpose into rknn model.