In model.attention AttentionModule1.shortcut_short is not used. You calculate the shortcut with the downsample weights. ```python shortcut_short = self.soft_resdown3(x_s) ``` AttentionModule3.shortcut_short is unnecessary.
In model.attention
AttentionModule1.shortcut_short is not used. You calculate the shortcut with the downsample weights.
AttentionModule3.shortcut_short is unnecessary.