Update metaformers.py

2 years ago · 5de2544a80
parent 55d4eb78a2
commit 5de2544a80
1 changed files with 1 additions and 0 deletions
--- 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)