Update davit.py

3 years ago · 2fbc436130
parent ba70ebc804
commit 2fbc436130
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -36,7 +36,7 @@ __all__ = ['DaViT']

 class MySequential(nn.Sequential):

-    def __init__(*args, **kwargs):
+    def __init__(self, *args, **kwargs):
        super(MySequential, self).__init__(*args, **kwargs)