Update tests for new mlp models

3 years ago · 6d81374b88
parent d5af752117
commit 6d81374b88
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -15,7 +15,7 @@ if hasattr(torch._C, '_jit_set_profiling_executor'):
    torch._C._jit_set_profiling_mode(False)

 # transformer models don't support many of the spatial / feature based model functionalities
-NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', 'mixer_*']
+NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*']
 NUM_NON_STD = len(NON_STD_FILTERS)

 # exclude models that cause specific test failures