From 5c73dd66cbed2fb5b51fef6da7a7100b076c1387 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 10 Dec 2022 04:42:09 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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():