在yolov5中加入了CloAttention注意力机制报错

[zqrxiaer]

我用的是YOLOv5s模型为基础，在backbone第九层加入了CloAttention，照着视频里的做法，修改了yolo.py文件，但是训练的时候报错：
"RuntimeError: shape '[28, 2, -1, 64, 24]' is invalid for input of size 286720"
错误发生在第77行：
“kv = to_kv(kv).view(b, 2, -1, self.dim_head, (h*w)//(self.window_size**2)).permute(1, 0, 2, 4, 3).contiguous() #(2 b m (H W) d)”
是维度错误，每一轮训练结束后的验证阶段开始的时候报错.