Open zqrxiaer opened 7 months ago
我用的是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)” 是维度错误，每一轮训练结束后的验证阶段开始的时候报错.
我用的是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)” 是维度错误，每一轮训练结束后的验证阶段开始的时候报错.