From 63970e4e611b42df91e7b8d2829f1c2d4e654814 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Mon, 29 Apr 2024 17:19:11 -0700 Subject: [PATCH] Update tests for volo/cait/xcit forward_intermediates --- tests/test_models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/test_models.py b/tests/test_models.py index 21f37a76..7f696dc1 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -49,10 +49,11 @@ if hasattr(torch._C, '_jit_set_profiling_executor'): # models with forward_intermediates() and support for FeatureGetterNet features_only wrapper FEAT_INTER_FILTERS = [ - 'vit_*', 'twins_*', 'deit*', 'beit*', 'mvitv2*', 'eva*', 'samvit_*', 'flexivit*' + 'vit_*', 'twins_*', 'deit*', 'beit*', 'mvitv2*', 'eva*', 'samvit_*', 'flexivit*', + 'cait_*', 'xcit_*', 'volo_*', ] -# transformer models don't support many of the spatial / feature based model functionalities +# transformer / hybrid models don't support full set of spatial / feature APIs and/or have spatial output. NON_STD_FILTERS = [ 'vit_*', 'tnt_*', 'pit_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*', 'convit_*', 'levit*', 'visformer*', 'deit*', 'xcit_*', 'crossvit_*', 'beit*',