diff --git a/timm/models/davit.py b/timm/models/davit.py index bef04e87..1f0055b6 100644 --- 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)