RuntimeError: shape '[32, 4, 1600]' is invalid for input of size 51200

iscyy / yoloair

🔥🔥🔥 专注于YOLOv5，YOLOv7、YOLOv8、YOLOv9改进模型，Support to improve backbone, neck, head, loss, IoU, NMS and other modules🚀

https://github.com/iscyy/yoloair

GNU General Public License v3.0

2.51k stars 430 forks source link

RuntimeError: shape '[32, 4, 1600]' is invalid for input of size 51200 #65

Closed Luxiangxianghy closed 2 years ago

Luxiangxianghy commented 2 years ago

Replace YOLOx-s backbone with MobileNetV3-small,the above-mentioned problems have occurred,I configured it according to yolov5-mobile3s.yaml, How do I solve it?

Luxiangxianghy commented 2 years ago

Resolved!

WenHui001 commented 1 year ago

How did you resolve it?