mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Missed comitting outstanding changes to default_cfg keys and test exclusions for swin v2

pull/1150/head
Ross Wightman 2022-02-23 19:50:26 -08:00
parent c6e4b7895a
commit 1420c118df
2 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_models.py
View File

@ -33,8 +33,9 @@ if 'GITHUB_ACTIONS' in os.environ:
EXCLUDE_FILTERS = [
'*efficientnet_l2*', '*resnext101_32x48d', '*in21k', '*152x4_bitm', '*101x3_bitm', '*50x3_bitm',
'*nfnet_f3*', '*nfnet_f4*', '*nfnet_f5*', '*nfnet_f6*', '*nfnet_f7*', '*efficientnetv2_xl*',
'*resnetrs350*', '*resnetrs420*', 'xcit_large_24_p8*', 'vit_huge*', 'vit_gi*']
NON_STD_EXCLUDE_FILTERS = ['vit_huge*', 'vit_gi*']
'*resnetrs350*', '*resnetrs420*', 'xcit_large_24_p8*', 'vit_huge*', 'vit_gi*', 'swin*huge*',
'swin*giant*']
NON_STD_EXCLUDE_FILTERS = ['vit_huge*', 'vit_gi*', 'swin*giant*']
else:
EXCLUDE_FILTERS = []
NON_STD_EXCLUDE_FILTERS = ['vit_gi*']

15
timm/models/swin_transformer_v2_cr.py
View File

@ -813,8 +813,7 @@ def swin_v2_cr_small_384(pretrained=False, **kwargs):
num_heads=(3, 6, 12, 24),
**kwargs
)
return _create_swin_transformer_v2_cr(
'swin_v2_cr_small_384', pretrained=pretrained, **model_kwargs
return _create_swin_transformer_v2_cr('swin_v2_cr_small_384', pretrained=pretrained, **model_kwargs
)
@ -863,8 +862,7 @@ def swin_v2_cr_large_384(pretrained=False, **kwargs):
num_heads=(6, 12, 24, 48),
**kwargs
)
return _create_swin_transformer_v2_cr(
'swin_v2_cr_large_384', pretrained=pretrained, **model_kwargs
return _create_swin_transformer_v2_cr('swin_v2_cr_large_384', pretrained=pretrained, **model_kwargs
)
@ -877,7 +875,7 @@ def swin_v2_cr_large_224(pretrained=False, **kwargs):
num_heads=(6, 12, 24, 48),
**kwargs
)
return _create_swin_transformer_v2_cr('swin_v2_cr_large_patch4_window7_224', pretrained=pretrained, **model_kwargs)
return _create_swin_transformer_v2_cr('swin_v2_cr_large_224', pretrained=pretrained, **model_kwargs)
@register_model
@ -890,7 +888,7 @@ def swin_v2_cr_huge_384(pretrained=False, **kwargs):
extra_norm_period=6,
**kwargs
)
return _create_swin_transformer_v2_cr('swin_v2_cr_huge_patch4_window12_384', pretrained=pretrained, **model_kwargs)
return _create_swin_transformer_v2_cr('swin_v2_cr_huge_384', pretrained=pretrained, **model_kwargs)
@register_model
@ -903,7 +901,7 @@ def swin_v2_cr_huge_224(pretrained=False, **kwargs):
extra_norm_period=6,
**kwargs
)
return _create_swin_transformer_v2_cr('swin_v2_cr_huge_patch4_window7_224', pretrained=pretrained, **model_kwargs)
return _create_swin_transformer_v2_cr('swin_v2_cr_huge_224', pretrained=pretrained, **model_kwargs)
@register_model
@ -916,8 +914,7 @@ def swin_v2_cr_giant_384(pretrained=False, **kwargs):
extra_norm_period=6,
**kwargs
)
return _create_swin_transformer_v2_cr(
'swin_v2_cr_giant_384', pretrained=pretrained, **model_kwargs
return _create_swin_transformer_v2_cr('swin_v2_cr_giant_384', pretrained=pretrained, **model_kwargs
)