Filter large vit models from torchscript tests

rwightman · rwightman · commit fd92ba0de85c · 2021-05-25T12:52:07.000-07:00
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -178,7 +178,8 @@ def test_model_features_pretrained(model_name, batch_size):
 
 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_*',
 ]
 
 

Original file line number	Diff line number	Diff line change
`@@ -178,7 +178,8 @@ def test_model_features_pretrained(model_name, batch_size):`
`178`	`178`
`179`	`179`	`EXCLUDE_JIT_FILTERS = [`
`180`	`180`	`'iabn', 'tresnet*', # models using inplace abn unlikely to ever be scriptable`
`181`		`- 'dla', 'hrnet', 'ghostnet*', # hopefully fix at some point`
	`181`	`+ 'dla', 'hrnet', 'ghostnet*', # hopefully fix at some point`
	`182`	`+ 'vit_large_', 'vit_huge_',`
`182`	`183`	`]`
`183`	`184`
`184`	`185`