diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 679d7ed6..ce5e99d3 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -363,7 +363,7 @@ class RandomMixing(nn.Module): x = x.reshape(B, H, W, C) return x - +''' class LayerNormGeneral(nn.Module): r""" General LayerNorm for different situations. @@ -462,7 +462,7 @@ class LayerNormGeneral(nn.Module): if self.use_bias: x = x + self.bias return x -''' + class SepConv(nn.Module): r""" Inverted separable convolution from MobileNetV2: https://arxiv.org/abs/1801.04381.