[Enhance] Improve CPE performance by reduce memory copy. (open-mmlab#762

)
mzr1996 · Apr 2, 2022 · 2473813 · 2473813
1 parent 8e2f05b
commit 2473813
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: