mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

fix img_size type

Browse Source
This commit is contained in:
Xihua Dong 2025-04-18 09:13:55 -07:00 committed by Ross Wightman
parent 3ff3899026
commit 0c136f7dab
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/layers/patch_embed.py
View File

@ -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,