Update swin.py

open-mmlab · Jun 18, 2022 · 605a55b · 605a55b
1 parent b8ece4a
commit 605a55b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmdet/models/backbones/swin.py b/mmdet/models/backbones/swin.py
@@ -753,8 +753,8 @@ def forward(self, x):
  mode='bicubic',
  align_corners=False).flatten(2).transpose(1, 2)
  else:
- absolute_pos_embed = self.absolute_pos_embed\
- .flatten(2).transpose(1, 2)
+ absolute_pos_embed = self.absolute_pos_embed.flatten(
+ 2).transpose(1, 2)
  x = x + absolute_pos_embed
  x = self.drop_after_pos(x)