[Fix] fix sam bug (#1633)

2023-06-15 10:10:51 +08:00 · 2023-06-15 10:10:51 +08:00 · 53648baca5
parent 3eaf719a64
commit 53648baca5
1 changed files with 3 additions and 3 deletions
--- a/mmpretrain/models/backbones/vit_sam.py
+++ b/mmpretrain/models/backbones/vit_sam.py
@ -593,11 +593,11 @@ class ViTSAM(BaseBackbone):

            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)