From cc8575e9d1da617040d28923450a9ee387b6fa11 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Sun, 8 Jan 2023 12:33:57 -0800
Subject: [PATCH] Update metaformers.py

---
 timm/models/metaformers.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/timm/models/metaformers.py b/timm/models/metaformers.py
index f909b079..2e6a59af 100644
--- a/timm/models/metaformers.py
+++ b/timm/models/metaformers.py
@@ -735,6 +735,7 @@ class MetaFormer(nn.Module):
 
     def forward(self, x):
         x = self.forward_features(x)
+        print(x.shape)
         x = self.forward_head(x)
         return x