From b745d30a3e770bf471db7d6ebae6ed702f7663bc Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 25 Oct 2021 15:15:14 -0700 Subject: [PATCH] Fix formatting of last commit --- timm/models/layers/patch_embed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/timm/models/layers/patch_embed.py b/timm/models/layers/patch_embed.py index b0b759c7..41528efa 100644 --- a/timm/models/layers/patch_embed.py +++ b/timm/models/layers/patch_embed.py @@ -30,8 +30,8 @@ class PatchEmbed(nn.Module): def forward(self, x): B, C, H, W = x.shape - torch._assert(H == self.img_size[0], f"Input image height ({H}) doesn't match model {self.img_size[0]}.") - torch._assert(W == self.img_size[1], f"Input image width ({W}) doesn't match model ({self.img_size[1]}.") + torch._assert(H == self.img_size[0], f"Input image height ({H}) doesn't match model ({self.img_size[0]}).") + torch._assert(W == self.img_size[1], f"Input image width ({W}) doesn't match model ({self.img_size[1]}).") x = self.proj(x) if self.flatten: x = x.flatten(2).transpose(1, 2) # BCHW -> BNC