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

Add fixed_input_size flag to msra efficient_vit

Browse Source
This commit is contained in:
Ross Wightman 2023-08-18 16:48:17 -07:00
parent c28324a150
commit 58ea1c02c4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/efficientvit_msra.py
View File

@ -518,6 +518,7 @@ def _cfg(url='', **kwargs):
'std': IMAGENET_DEFAULT_STD, 'std': IMAGENET_DEFAULT_STD,
'first_conv': 'patch_embed.conv1.conv', 'first_conv': 'patch_embed.conv1.conv',
'classifier': 'head.linear', 'classifier': 'head.linear',
'fixed_input_size': True,
**kwargs, **kwargs,
} }