From 6d81374b88ba7390cb5045b512787c6cc5b728f1 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 19 May 2021 11:09:42 -0700 Subject: [PATCH] Update tests for new mlp models --- tests/test_models.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index ced2fd76..b77b29ff 100644 --- 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