From 809271b0f3e5d9be4e11c0c5cec1dbba8b5e2c60 Mon Sep 17 00:00:00 2001 From: contrastive Date: Mon, 29 Mar 2021 09:39:50 +0800 Subject: [PATCH] Update test_models.py --- 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 5182c998..8e2b674e 100644 --- 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():