diff --git a/timm/layers/mlp.py b/timm/layers/mlp.py index 2c307330..a68ae4bc 100644 --- a/timm/layers/mlp.py +++ b/timm/layers/mlp.py @@ -42,6 +42,7 @@ class Mlp(nn.Module): x = self.fc1(x) x = self.act(x) x = self.drop1(x) + x = self.norm(x) x = self.fc2(x) x = self.drop2(x) return x