Revert "Update metaformers.py"

This reverts commit 2fef9006d72824789b733dd8dbc20b21b79add0b.
This commit is contained in:
Fredo Guan 2023-01-17 12:34:29 -08:00
parent 869e01cec6
commit 47b718a1c8

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)
return x