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)