From 66929838320385d3e0c7638b2a6ad937107a2d7d Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Tue, 6 Dec 2022 20:35:07 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index da4d7f52..72aefb00 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -38,13 +38,13 @@ __all__ = ['DaViT'] class MySequential(nn.Sequential): - @Overload + @overload def forward(self, inputs : Tensor): for module in self._modules.values(): inputs = module(inputs) return inputs - @Overload + @overload def forward(self, inputs : Tuple[Tensor, Tensor]): for module in self._modules.values(): inputs = module(*inputs)