提交 5ab372a3 编写于 作者: R Ross Wightman

Merge branch 'master' of https://github.com/abcdvzz/pytorch-image-models into abcdvzz-master

......@@ -15,7 +15,7 @@ if hasattr(torch._C, '_jit_set_profiling_executor'):
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_*']
NON_STD_FILTERS = ['vit_*', 'tnt_*', 'pit_*', 'swin_*', 'coat_*', 'cait_*', '*mixer_*', 'gmlp_*', 'resmlp_*', 'twins_*']
NUM_NON_STD = len(NON_STD_FILTERS)
# exclude models that cause specific test failures
......
......@@ -39,6 +39,7 @@ from .vision_transformer_hybrid import *
from .vovnet import *
from .xception import *
from .xception_aligned import *
from .twins import *
from .factory import create_model, split_model_name, safe_model_name
from .helpers import load_checkpoint, resume_checkpoint, model_parameters
......
此差异已折叠。
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册