diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index 1eea2129..be03eec3 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -414,7 +414,7 @@ class LayerNormGeneral(nn.Module): x = x * self.weight if self.use_bias: x = x + self.bias - return + return x class SepConv(nn.Module): r"""