From daac49c9a107778c95e11a16fd5b4a8b45513678 Mon Sep 17 00:00:00 2001 From: Longguang Wang Date: Mon, 18 Oct 2021 15:36:56 +0800 Subject: [PATCH] fix a bug --- model/smsr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model/smsr.py b/model/smsr.py index 224918e..97e6c02 100644 --- 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)