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)