Update davit.py

3 years ago · 9ad97c9fb1
parent 2faeaebfee
commit 9ad97c9fb1
1 changed files with 1 additions and 1 deletions
--- 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)