Update test_models.py

2021-03-29 09:39:50 +08:00 · 2021-03-29 09:39:50 +08:00 · 809271b0f3
parent de86314655
commit 809271b0f3
1 changed files with 1 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -14,7 +14,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_*']
+NON_STD_FILTERS = ['vit_*', 'tnt_*']

 # exclude models that cause specific test failures
 if 'GITHUB_ACTIONS' in os.environ:  # and 'Linux' in platform.system():