From d8cce7d1e46e81916e1b93f8b65b9e900916f024 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 07:48:43 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index 7442409f..cee1b404 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -525,8 +525,8 @@ class DaViT(nn.Module): class DaViTFeatures(DaViT): def __init__(self, *args, **kwargs): - super(DaViT, self).__init__(*args, **kwargs) - self.feature_info = FeatureInfo(self.feature_info, kwargs.get('out_inices', (1, 2, 3, 4))) + super().__init__(*args, **kwargs) + self.feature_info = FeatureInfo(self.feature_info, kwargs.get('out_inices', (0, 1, 2, 3))) def forward(self, x) -> List[Tensor]: return self.forward_pyramid_features(x)