Update dependencyvit.py

huggingface · Dec 7, 2024 · d0d96db · d0d96db
1 parent b7b5073
commit d0d96db
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)