From 05092e2fbeafa034b363094876c54eec04c342ad Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 20 Nov 2021 15:51:48 -0800 Subject: [PATCH] Add more models to FX filter --- tests/test_models.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/test_models.py b/tests/test_models.py index 5fde43da..f55247ee 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -337,7 +337,12 @@ def _create_fx_model(model, train=False): EXCLUDE_FX_FILTERS = [] # not enough memory to run fx on more models than other tests if 'GITHUB_ACTIONS' in os.environ: - EXCLUDE_FX_FILTERS += ['beit_large*', 'swin_large*'] + EXCLUDE_FX_FILTERS += [ + 'beit_large*', + 'swin_large*', + '*resnext101_32x32d', + 'resnetv2_152x2*', + ] @pytest.mark.timeout(120)