[Fix] fix sam bug (open-mmlab#1633)

ASHORE1225 · Jun 15, 2023 · 53648ba · 53648ba
1 parent 3eaf719
commit 53648ba
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mmpretrain/models/backbones/vit_sam.py b/mmpretrain/models/backbones/vit_sam.py
@@ -593,11 +593,11 @@ def forward(self, x: torch.Tensor) -> Tuple[torch.Tensor]:
 
             if i in self.out_indices:
                 # (B, H, W, C) -> (B, C, H, W)
-                x = x.permute(0, 3, 1, 2)
+                x_reshape = x.permute(0, 3, 1, 2)
 
                 if self.out_channels > 0:
-                    x = self.channel_reduction(x)
-                outs.append(self._format_output(x))
+                    x_reshape = self.channel_reduction(x_reshape)
+                outs.append(self._format_output(x_reshape))
 
         return tuple(outs)