Update davit.py

3 years ago · 2979ad9113
parent bff156994e
commit 2979ad9113
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -524,7 +524,7 @@ class DaViT(nn.Module):
 class DaViTFeatures(DaViT):
-    def __init__(*args):
+    def __init__(*args, **kwargs):
        super(DaViT, self).__init__(*args, **kwargs)
        self.feature_info = FeatureInfo(self.feature_info, kwargs.get('out_inices', (1, 2, 3, 4)))