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

2022-04-02 11:12:20 +08:00 · 2022-04-02 11:12:20 +08:00 · 875195efe9
parent 4a3f8d02c5
commit 875195efe9
1 changed files with 1 additions and 1 deletions
--- a/mmcls/models/utils/position_encoding.py
+++ b/mmcls/models/utils/position_encoding.py
@ -32,7 +32,7 @@ class ConditionalPositionEncoding(BaseModule):
        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: