diff --git a/timm/models/davit.py b/timm/models/davit.py index 8ffde94c..4b04d535 100644 --- 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)