diff --git a/timm/models/davit.py b/timm/models/davit.py index 87007086..44d2c365 100644 --- 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():