Update davit.py

3 years ago · 5c73dd66cb
parent ab1faea450
commit 5c73dd66cb
1 changed files with 1 additions and 1 deletions
--- a/timm/models/davit.py
+++ b/timm/models/davit.py
@ -45,7 +45,7 @@ class SequentialWithSize(nn.Sequential):

 class SequentialWithSize(nn.Sequential):
    def __init__(self, *args, **kwargs):
-        super(SequentialWithSize, self).__init__(*args, **kwargs)
+        super(SequentialWithSize, nn.Sequential, self).__init__(*args, **kwargs)
        
    def forward(self, x : Tensor, size: Tuple[int, int]):
        for module in self._modules.values():