From 53e66d223f0072558b2bf8426e208c01e4398f31 Mon Sep 17 00:00:00 2001 From: Fredo Guan Date: Sat, 10 Dec 2022 21:25:09 -0800 Subject: [PATCH] Update davit.py --- timm/models/davit.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/timm/models/davit.py b/timm/models/davit.py index 94e1bc25..ff3442d6 100644 --- a/timm/models/davit.py +++ b/timm/models/davit.py @@ -165,11 +165,7 @@ class PatchEmbed(nn.Module): if self.norm.normalized_shape[0] == self.in_chans: x = self.norm(x.permute(0, 2, 3, 1)).permute(0, 3, 1, 2) - print(self.patch_size[1] - W % self.patch_size[1]) - print(self.patch_size[0] - H % self.patch_size[0]) - x = F.pad(x, (0, (self.patch_size[1] - W % self.patch_size[1]) % self.patch_size[1])) - x = F.pad(x, (0, 0, 0, (self.patch_size[0] - H % self.patch_size[0]) % self.patch_size[0])) x = self.proj(x)