Skip to content

Commit a897e0e

Browse files
committed
Merge branch 'feature/crossvit' of https://github.com/chunfuchen/pytorch-image-models into chunfuchen-feature/crossvit
2 parents 54e90e8 + 9fe5798 commit a897e0e

File tree

3 files changed

+463
-1
lines changed

3 files changed

+463
-1
lines changed

tests/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
# transformer models don't support many of the spatial / feature based model functionalities
1818
NON_STD_FILTERS = [
1919
'vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*',
20-
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*']
20+
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*']
2121
NUM_NON_STD = len(NON_STD_FILTERS)
2222

2323
# exclude models that cause specific test failures

timm/models/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from .byoanet import *
22
from .byobnet import *
33
from .cait import *
4+
from .crossvit import *
45
from .coat import *
56
from .convit import *
67
from .cspnet import *

0 commit comments

Comments
 (0)