From 4a936b92a903235273903c1ffe8928acc4cbc1b6 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 02:10:41 -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 cad176fa..7834546c 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, sizes
+        return features[:-1], sizes[:-1]
     
     def forward_pyramid_features(self, x):
         x, sizes = self.forward_network(x)