Skip to content

Commit cb294c8

Browse files
committed
Update tests for aimv2 filtering
1 parent 1d6ebeb commit cb294c8

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

tests/test_models.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -53,13 +53,13 @@
5353
'vision_transformer', 'vision_transformer_sam', 'vision_transformer_hybrid', 'vision_transformer_relpos',
5454
'beit', 'mvitv2', 'eva', 'cait', 'xcit', 'volo', 'twins', 'deit', 'swin_transformer', 'swin_transformer_v2',
5555
'swin_transformer_v2_cr', 'maxxvit', 'efficientnet', 'mobilenetv3', 'levit', 'efficientformer', 'resnet',
56-
'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit', 'hieradet_sam2'
56+
'regnet', 'byobnet', 'byoanet', 'mlp_mixer', 'hiera', 'fastvit', 'hieradet_sam2', 'aimv2*'
5757
]
5858

5959
# transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output.
6060
NON_STD_FILTERS = [
6161
'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
62-
'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*',
62+
'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*', 'aimv2*',
6363
'poolformer_*', 'volo_*', 'sequencer2d_*', 'mvitv2*', 'gcvit*', 'efficientformer*', 'sam_hiera*',
6464
'eva_*', 'flexivit*', 'eva02*', 'samvit_*', 'efficientvit_m*', 'tiny_vit_*', 'hiera_*', 'vitamin*', 'test_vit*',
6565
]
@@ -73,10 +73,10 @@
7373
'*nfnet_f3*', '*nfnet_f4*', '*nfnet_f5*', '*nfnet_f6*', '*nfnet_f7*', '*efficientnetv2_xl*',
7474
'*resnetrs350*', '*resnetrs420*', 'xcit_large_24_p8*', '*huge*', '*giant*', '*gigantic*',
7575
'*enormous*', 'maxvit_xlarge*', 'regnet*1280', 'regnet*2560']
76-
NON_STD_EXCLUDE_FILTERS = ['*huge*', '*giant*', '*gigantic*', '*enormous*']
76+
NON_STD_EXCLUDE_FILTERS = ['*huge*', '*giant*', '*gigantic*', '*enormous*', '*_1b_*', '*_3b_*']
7777
else:
7878
EXCLUDE_FILTERS = ['*enormous*']
79-
NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*']
79+
NON_STD_EXCLUDE_FILTERS = ['*gigantic*', '*enormous*', '*_3b_*']
8080

8181
EXCLUDE_JIT_FILTERS = ['hiera_*']
8282

timm/models/vision_transformer.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2242,6 +2242,8 @@ def _cfg(url: str = '', **kwargs) -> Dict[str, Any]:
22422242
'test_vit3.r160_in1k': _cfg(
22432243
hf_hub_id='timm/',
22442244
input_size=(3, 160, 160), crop_pct=0.95),
2245+
'test_vit4.r160_in1k': _cfg(
2246+
input_size=(3, 160, 160), crop_pct=0.95),
22452247
}
22462248

22472249
_quick_gelu_cfgs = [n for n, c in default_cfgs.items() if c.get('notes', ()) and 'quickgelu' in c['notes'][0]]

0 commit comments

Comments
 (0)