关于C3ResAtnMHSA模块的问题

[SwjtuerFz]

师兄您好！有幸阅读了您的论文。想请教两个问题： 一、由于我尝试过直接使用MHSA加入C3中，然后分别将C3-MHSA模块加入到和您的网络差不多的位置，但是发现显存占用很大。而您的模块加了那么多个地方是因为每次使用都把通道数降低了的，从而可以降低显存占用是吗？如下图，head中第一个C3ResAtnMHSA模块，输入512通道，输出32...... 二、我如果直接使用您这个C3ResAtnMHSA模块将入到我的yolov5s改进的网络中，除了修改common和yolo的那两个文件以外，还有别的地方需要修改吗？我的输入图像是768大小的，就会报错。