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)