Update davit.py

3 years ago · 5a9f582271
parent 8ac5ec2d88
commit 5a9f582271
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -384,7 +384,7 @@ class DaViT(nn.Module):
        num_classes=1000,
        global_pool='avg'#,
        #features_only = False
-    ):
+    **kwargs):
        super().__init__()

        architecture = [[index] * item for index, item in enumerate(depths)]