yolov5 test.py "cannot reshape array"问题求助

rknn-toolkit-1.7.1，example中的pytorch yolov5 test.py，实际测试出现如下问题，请大牛们指点，搞了两天了都没搞定，谢谢
--> Init runtime environment
librknn_runtime version 1.7.1 (bd41dbc build: 2021-10-28 16:15:23 base: 1131)
done
--> Running model
Warning: Cmodel got negative postshift!
Warning: Cmodel got negative postshift!
Warning: Cmodel got negative postshift!
Traceback (most recent call last):
  File "test.py", line 294, in <module>
    input1_data = input1_data.reshape([3,-1]+list(input1_data.shape[-2:]))
ValueError: cannot reshape array of size 409600 into shape (3,newaxis,40,40)