CBAM复现代码问题

CBAM复现代码77-79三行:
CBAM_out = self.channel(out)
CBAM_out = self.spatial(CBAM_out)
out = out * CBAM_out
按原论文,似应为:
CBAM_out = self.channel(out)*out
CBAM_out = self.spatial(CBAM_out)*CBAM_out
不知对否?

问题相关代码,请勿粘贴截图
运行结果及报错内容
我的解答思路和尝试过的方法
我想要达到的结果