[Fix] Fix performance issue in convnext DDP train. (#1098)

to fix performance issue in convnext DDP train
open-mmlab · Oct 17, 2022 · 38040d5 · 38040d5
1 parent bcadb74
commit 38040d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmcls/models/backbones/convnext.py b/mmcls/models/backbones/convnext.py
@@ -36,8 +36,8 @@ def forward(self, x):
         assert x.dim() == 4, 'LayerNorm2d only supports inputs with shape ' \
             f'(N, C, H, W), but got tensor with shape {x.shape}'
         return F.layer_norm(
-            x.permute(0, 2, 3, 1), self.normalized_shape, self.weight,
-            self.bias, self.eps).permute(0, 3, 1, 2)
+            x.permute(0, 2, 3, 1).contiguous(), self.normalized_shape,
+            self.weight, self.bias, self.eps).permute(0, 3, 1, 2).contiguous()
 
 
 class ConvNeXtBlock(BaseModule):