Update README.md

README.md

@@ -56,6 +56,11 @@ with torch.no_grad():
     )(img).unsqueeze(0)
     img_tensor = img_tensor.to(device)
 
+    if img_tensor.shape[2] % 8 != 0:
+        img_tensor = img_tensor[:, :, : -(img_tensor.shape[2] % 8), :]
+    if img_tensor.shape[3] % 8 != 0:
+        img_tensor = img_tensor[:, :, :, : -(img_tensor.shape[3] % 8)]
+
     output = model(img_tensor)
 
 ```