From 5de2544a8025fbf2f71d4a33b384e2a59de32c76 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 17 Jan 2023 10:49:34 -0800 Subject: [PATCH] Update metaformers.py --- timm/models/metaformers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py index d96d6de9..4e63c1b8 100644 --- a/timm/models/metaformers.py +++ b/timm/models/metaformers.py @@ -536,6 +536,7 @@ class Mlp(nn.Module): self.drop2 = nn.Dropout(drop_probs[1]) def forward(self, x): + print(x) x = self.fc1(x) x = self.act(x) x = self.drop1(x)