Revert "Update metaformers.py"

This reverts commit 3f0b075367.
pull/1647/head
Fredo Guan 2 years ago
parent f985f35b91
commit 869e01cec6

@ -822,7 +822,7 @@ class MetaFormer(nn.Module):
#x = self.stages(x)
for i, stage in enumerate(self.stages):
x=stage(x)
#print(x[0][0][0][0])
print(x[0][0][0][0])
return x

Loading…
Cancel
Save