|
|
@ -15,7 +15,7 @@ if hasattr(torch._C, '_jit_set_profiling_executor'):
|
|
|
|
torch._C._jit_set_profiling_mode(False)
|
|
|
|
torch._C._jit_set_profiling_mode(False)
|
|
|
|
|
|
|
|
|
|
|
|
# transformer models don't support many of the spatial / feature based model functionalities
|
|
|
|
# transformer models don't support many of the spatial / feature based model functionalities
|
|
|
|
NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*']
|
|
|
|
NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*']
|
|
|
|
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
|
|
|
|