This reverts commit 2916f37f8d.
2916f37f8d
@ -816,10 +816,8 @@ class MetaFormer(nn.Module):
def forward_features(self, x):
x = self.patch_embed(x)
#x = self.stages(x)
x = self.stages(x)
for i, stage in enumerate(self.stages):
x = stage(x)
print(x[0][0][0][0])
return x