From 36527b5a33af077cc51b289fa7c8f939d3af36be Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Wed, 7 Dec 2022 01:08:25 -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 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)