fix img_size type

Xihua Dong · rwightman · commit 0c136f7dab5b · 2025-04-18T21:05:05.000-07:00
diff --git a/timm/layers/patch_embed.py b/timm/layers/patch_embed.py
@@ -31,7 +31,7 @@ class PatchEmbed(nn.Module):
 
     def __init__(
             self,
-            img_size: Optional[int] = 224,
+            img_size: Union[int, Tuple[int, int]] = 224,
             patch_size: int = 16,
             in_chans: int = 3,
             embed_dim: int = 768,
