diff --git a/timm/models/davit.py b/timm/models/davit.py index 40ed83b2..7442409f 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -524,8 +524,8 @@ class DaViT(nn.Module): class DaViTFeatures(DaViT): - def __init__(*args, **kwargs): - super(self).__init__(*args, **kwargs) + 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))) def forward(self, x) -> List[Tensor]: