yolov5在添加CA注意力机制模块的训练过程中报错特征尺度不匹配

问题遇到的现象和发生背景

想再yolov5网络模块中添加CA注意力机制模块
yolo.py可以跑通，但train.py的计算map过程中出现特征尺度不匹配，希望得到[32, 128, 168 168]却得到[32, 128, 112, 168]，改变bitchsize时得到[32, 128, 128, 168]

问题相关代码，请勿粘贴截图

下为train.py报错图片

下为网络模块添加图

下为网络结构配置文件

运行结果及报错内容

我的解答思路和尝试过的方法

我想要达到的结果