From bb4d5f74f7a85c5ac9631c8aab4ccb3620d60ce1 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 10 Dec 2022 04:48:14 -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 8a1ab8a7..d74f7361 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -37,7 +37,7 @@ __all__ = ['DaViT'] # modified nn.Sequential that includes a size tuple in the forward function class SequentialWithSize(nn.Sequential): -@torch.jit.ignore + @torch.jit.ignore def forward(self, x : Tensor, size: Tuple[int, int]): for module in self._modules.values(): x, size = module(x, size)