Update davit.py

3 years ago · 36527b5a33
parent f8c5150fcd
commit 36527b5a33
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -572,7 +572,7 @@ class DaViT(nn.Module):
        
        
        # non-normalized pyramid features + corresponding sizes
-        return tuple(features), tuple(sizes)
+        return features, sizes
        
    def forward_features(self, x):
        x, sizes = self.forward_features_full(x)