From 128d9729c3ef710dfc3d1f28b5e201512bd73e6e Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 6 Dec 2022 21:47:40 -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 e9957f19..fe3cbad4 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -36,7 +36,7 @@ from .registry import register_model __all__ = ['DaViT'] class MySequential(nn.Sequential): - def forward(self, inputs : Tuple[Tensor, Tensor]): + def forward(self, inputs : Tuple[Tensor, Tuple[int, int]]): for module in self._modules.values(): #if type(inputs) == tuple: inputs = module(*inputs)