RuntimeError: The size of tensor a (16) must match the size of tensor b (4) at non-singleton dimension 1

[huranzhijian]

My own imageSize is 1280 * 10, and it occured when I train my datesets.

[huranzhijian]

@cv516Buaa The yaml that i uses is yolov5l-xs-tr-cbam-spp-bifpn.

[rothfa]

@huranzhijian this worked for me https://blog.csdn.net/qq_55621259/article/details/125943353

You can also try to change the Backbone and add cat to the head.