Skip to content

Commit 1c9284c

Browse files
committed
Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now
1 parent f8a215c commit 1c9284c

File tree

3 files changed

+422
-1
lines changed

3 files changed

+422
-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_*', 'crossvit_*']
20+
'convit_*', 'levit*', 'visformer*', 'deit*', 'jx_nest_*', 'nest_*', 'xcit_*', 'crossvit_*', 'beit_*']
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,3 +1,4 @@
1+
from .beit import *
12
from .byoanet import *
23
from .byobnet import *
34
from .cait import *

0 commit comments

Comments
 (0)