From 8ef244296d87ae00ab7e3ed353b88f41eaa734f4 Mon Sep 17 00:00:00 2001 From: Yosua Michael Maranatha Date: Thu, 15 Sep 2022 12:11:08 -0700 Subject: [PATCH] [FBcode->GH]Fix vit model assert message to be compatible with torchmultimodal test --- torchvision/models/vision_transformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchvision/models/vision_transformer.py b/torchvision/models/vision_transformer.py index 699e64ae0c1..be62ce1ce96 100644 --- a/torchvision/models/vision_transformer.py +++ b/torchvision/models/vision_transformer.py @@ -268,8 +268,8 @@ def __init__( def _process_input(self, x: torch.Tensor) -> torch.Tensor: n, c, h, w = x.shape p = self.patch_size - torch._assert(h == self.image_size, f"Wrong image height, expected {self.image_size} but got {h}!") - torch._assert(w == self.image_size, f"Wrong image width, expected {self.image_size} but got {w}!") + torch._assert(h == self.image_size, f"Wrong image height! Expected {self.image_size} but got {h}!") + torch._assert(w == self.image_size, f"Wrong image width! Expected {self.image_size} but got {w}!") n_h = h // p n_w = w // p