diff --git a/timm/models/davit.py b/timm/models/davit.py index 49b8cfa2..2edc44eb 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -525,7 +525,7 @@ class DaViT(nn.Module): class DaViTFeatures(DaViT): def __init__(*args, **kwargs): - super().__init__(*args, **kwargs) + super(DaViT).__init__(*args, **kwargs) self.feature_info = FeatureInfo(self.feature_info, kwargs.get('out_inices', (1, 2, 3, 4))) def forward(self, x) -> List[Tensor]: