Add fixed_input_size flag to msra efficient_vit

2023-08-18 16:48:17 -07:00 · 2023-08-18 16:48:17 -07:00 · 58ea1c02c4
parent c28324a150
commit 58ea1c02c4
1 changed files with 1 additions and 0 deletions
--- a/timm/models/efficientvit_msra.py
+++ b/timm/models/efficientvit_msra.py
@ -518,6 +518,7 @@ def _cfg(url='', **kwargs):
        'std': IMAGENET_DEFAULT_STD,
        'first_conv': 'patch_embed.conv1.conv',
        'classifier': 'head.linear',
+        'fixed_input_size': True,
        **kwargs,
    }