Revert "Update metaformers.py"

This reverts commit 1d882eb494.
pull/1647/head
Fredo Guan 2 years ago
parent e4f89b2e25
commit 4f6bbbfb0e

@ -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)

Loading…
Cancel
Save