From d0d96db97304dc98aaf3364d66de3c1c0493a72d Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 7 Dec 2024 11:31:26 -0700 Subject: [PATCH] Update dependencyvit.py --- timm/models/dependencyvit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/dependencyvit.py b/timm/models/dependencyvit.py index a6711fc85..ee6b6d3aa 100644 --- a/timm/models/dependencyvit.py +++ b/timm/models/dependencyvit.py @@ -149,7 +149,7 @@ def __init__( dim: int, ) -> None: super().__init__() - self.proj = nn.Conv2d(dim, dim, 3, groups=dim) + self.proj = nn.Conv2d(dim, dim, 3, padding=1, groups=dim) def forward(self, x: torch.Tensor) -> torch.Tensor: return x + self.proj(x)