Skip to content

Commit 6e04da0

Browse files
committed
Merge branch 'abcdvzz-master'
2 parents b4ebf92 + a569635 commit 6e04da0

File tree

3 files changed

+433
-1
lines changed

3 files changed

+433
-1
lines changed

tests/test_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
torch._C._jit_set_profiling_mode(False)
1616

1717
# transformer models don't support many of the spatial / feature based model functionalities
18-
NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*']
18+
NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*']
1919
NUM_NON_STD = len(NON_STD_FILTERS)
2020

2121
# 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
@@ -39,6 +39,7 @@
3939
from .vovnet import *
4040
from .xception import *
4141
from .xception_aligned import *
42+
from .twins import *
4243

4344
from .factory import create_model, split_model_name, safe_model_name
4445
from .helpers import load_checkpoint, resume_checkpoint, model_parameters

0 commit comments

Comments
 (0)