Revert "Update metaformers.py"

This reverts commit 3f0b07536764c47af42290a218a15097b5cec1a0.
This commit is contained in:
Fredo Guan 2023-01-17 12:34:01 -08:00
parent f985f35b91
commit 869e01cec6

View File

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