mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Fix gc effv2 model cfg name

Browse Source
This commit is contained in:
Ross Wightman 2021-07-11 12:14:31 -07:00
parent 392368e210
commit f8039c7492
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/efficientnet.py
View File

@ -1505,7 +1505,7 @@ def efficientnetv2_rw_t(pretrained=False, **kwargs):
def gc_efficientnetv2_rw_t(pretrained=False, **kwargs): def gc_efficientnetv2_rw_t(pretrained=False, **kwargs):
""" EfficientNet-V2 Tiny w/ Global Context Attn (Custom variant, tiny not in paper). """ """ EfficientNet-V2 Tiny w/ Global Context Attn (Custom variant, tiny not in paper). """
model = _gen_efficientnetv2_s( model = _gen_efficientnetv2_s(
'gc_efficientnetv2_t', channel_multiplier=0.8, depth_multiplier=0.9, 'gc_efficientnetv2_rw_t', channel_multiplier=0.8, depth_multiplier=0.9,
rw=False, se_layer='gc', pretrained=pretrained, **kwargs) rw=False, se_layer='gc', pretrained=pretrained, **kwargs)
return model return model