From aa219391e403566ada72db362b28ff68763a7e91 Mon Sep 17 00:00:00 2001
From: Fredo Guan <fredo.guan@hotmail.com>
Date: Thu, 8 Dec 2022 07:38:26 -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 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]: