fix a bug

The-Learning-And-Vision-Atelier-LAVA · Oct 18, 2021 · daac49c · daac49c
1 parent af9b04c
commit daac49c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/model/smsr.py b/model/smsr.py
@@ -250,8 +250,8 @@ def forward(self, x):
                     fea = self.body[i](fea)
                     fea = fea * ch_mask[:, :, i:i + 1, 1:] * spa_mask + fea * ch_mask[:, :, i:i + 1, :1]
                 else:
-                    fea_d = self.body[i](fea * ch_mask[:, :, i:i + 1, :1])
-                    fea_s = self.body[i](fea * ch_mask[:, :, i:i + 1, 1:])
+                    fea_d = self.body[i](fea * ch_mask[:, :, i - 1:i, :1])
+                    fea_s = self.body[i](fea * ch_mask[:, :, i - 1:i, 1:])
                     fea = fea_d * ch_mask[:, :, i:i + 1, 1:] * spa_mask + fea_d * ch_mask[:, :, i:i + 1, :1] + \
                           fea_s * ch_mask[:, :, i:i + 1, 1:] * spa_mask + fea_s * ch_mask[:, :, i:i + 1, :1] * spa_mask
                 fea = self.relu(fea)