diff --git a/tests/test_models.py b/tests/test_models.py index 4e50de6e..f06ddd95 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -25,7 +25,7 @@ if hasattr(torch._C, '_jit_set_profiling_executor'): NON_STD_FILTERS = [ 'vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*', 'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit_*', - 'poolformer_*', 'volo_*'] + 'poolformer_*', 'volo_*', 'sequencer2d_*'] NUM_NON_STD = len(NON_STD_FILTERS) # exclude models that cause specific test failures