Filter large vit models from torchscript tests

3 years ago · fd92ba0de8
parent 318360c3f9
commit fd92ba0de8
1 changed files with 2 additions and 1 deletions
--- a/tests/test_models.py
+++ b/tests/test_models.py
@ -178,7 +178,8 @@ if 'GITHUB_ACTIONS' not in os.environ:

 EXCLUDE_JIT_FILTERS = [
    '*iabn*', 'tresnet*',  # models using inplace abn unlikely to ever be scriptable
-    'dla*', 'hrnet*', 'ghostnet*', # hopefully fix at some point
+    'dla*', 'hrnet*', 'ghostnet*',  # hopefully fix at some point
+    'vit_large_*', 'vit_huge_*',
 ]