From f8039c7492784099db08ac19a4da630aba5d248d Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 11 Jul 2021 12:14:31 -0700 Subject: [PATCH] Fix gc effv2 model cfg name --- timm/models/efficientnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 1baf9d91..848aeddd 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1505,7 +1505,7 @@ def 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). """ 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) return model