Update davit.py

timm/models/davit.py

@@ -35,6 +35,11 @@ __all__ = ['DaViT']
 
 
 class MySequential(nn.Sequential):
+
+    def __init__(*args, **kwargs):
+        super(MySequential, self).__init__(*args, **kwargs)
+
+
     def forward(self, inputs : Tensor, size : Tuple[int, int]):
         for module in self._modules.values():
             inputs = module(inputs, size)