Update davit.py

3 years ago · 54f74473d2
parent a59189a48d
commit 54f74473d2
1 changed files with 2 additions and 3 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -35,9 +35,8 @@ from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
 __all__ = ['DaViT']

 class SequentialWithSize(nn.Sequential):
-
-    def __init__(*args, **kwargs):
-        super(MySequential, self).__init__(*args, **kwargs)
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)

    def forward(self, x: Tensor, size: Tuple[int, int]):
        for module in self._modules.values():