Update metaformers.py

pull/1647/head
Fredo Guan 2 years ago
parent 2209d0830e
commit 1d882eb494

@ -536,7 +536,6 @@ 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