From d844cd5bd6d20f0d68cfdeb0ac0091a77c8cf9d0 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Thu, 8 Dec 2022 07:39:15 -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 2edc44eb..40ed83b2 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(DaViT).__init__(*args, **kwargs) + super(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]: