[Enhance] Improve CPE performance by reduce memory copy. (#762)

open-mmlab · Apr 2, 2022 · 875195e · 875195e
1 parent 4a3f8d0
commit 875195e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcls/models/utils/position_encoding.py b/mmcls/models/utils/position_encoding.py
@@ -32,7 +32,7 @@ def forward(self, x, hw_shape):
         H, W = hw_shape
         feat_token = x
         # convert (B, N, C) to (B, C, H, W)
-        cnn_feat = feat_token.transpose(1, 2).view(B, C, H, W)
+        cnn_feat = feat_token.transpose(1, 2).view(B, C, H, W).contiguous()
         if self.stride == 1:
             x = self.proj(cnn_feat) + cnn_feat
         else: