|
|
@ -25,7 +25,7 @@ if hasattr(torch._C, '_jit_set_profiling_executor'):
|
|
|
|
NON_STD_FILTERS = [
|
|
|
|
NON_STD_FILTERS = [
|
|
|
|
'vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
|
|
|
|
'vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
|
|
|
|
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit_*',
|
|
|
|
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit_*',
|
|
|
|
'poolformer_*', 'volo_*']
|
|
|
|
'poolformer_*', 'volo_*', 'sequencer2d_*']
|
|
|
|
NUM_NON_STD = len(NON_STD_FILTERS)
|
|
|
|
NUM_NON_STD = len(NON_STD_FILTERS)
|
|
|
|
|
|
|
|
|
|
|
|
# exclude models that cause specific test failures
|
|
|
|
# exclude models that cause specific test failures
|
|
|
|