Merge branch 'convit' of https://github.com/amaarora/pytorch-image-models into amaarora-convit

Author: rwightman

tests/test_models.py

@@ -15,7 +15,8 @@
     torch._C._jit_set_profiling_mode(False)
 
 # transformer models don't support many of the spatial / feature based model functionalities
-NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*']
+NON_STD_FILTERS = [
+    'vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*', 'convit_*']
 NUM_NON_STD = len(NON_STD_FILTERS)
 
 # exclude models that cause specific test failures

timm/models/__init__.py

@@ -2,6 +2,7 @@
 from .byobnet import *
 from .cait import *
 from .coat import *
+from .convit import *
 from .cspnet import *
 from .densenet import *
 from .dla import *