From 9ad97c9fb1debaa960d9564cff7c0795ff814419 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 02:04:09 -0800
Subject: [PATCH] Update davit.py

---
 timm/models/davit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/timm/models/davit.py b/timm/models/davit.py
index 15429003..cad176fa 100644
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@@ -600,7 +600,7 @@ class DaViT(nn.Module):
         
         
         # non-normalized pyramid features + corresponding sizes
-        return features[1:], sizes[:-1]
+        return features, sizes
     
     def forward_pyramid_features(self, x):
         x, sizes = self.forward_network(x)