Skip to content

Commit f8039c7

Browse files
committed
Fix gc effv2 model cfg name
1 parent 392368e commit f8039c7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/efficientnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1505,7 +1505,7 @@ def efficientnetv2_rw_t(pretrained=False, **kwargs):
15051505
def gc_efficientnetv2_rw_t(pretrained=False, **kwargs):
15061506
""" EfficientNet-V2 Tiny w/ Global Context Attn (Custom variant, tiny not in paper). """
15071507
model = _gen_efficientnetv2_s(
1508-
'gc_efficientnetv2_t', channel_multiplier=0.8, depth_multiplier=0.9,
1508+
'gc_efficientnetv2_rw_t', channel_multiplier=0.8, depth_multiplier=0.9,
15091509
rw=False, se_layer='gc', pretrained=pretrained, **kwargs)
15101510
return model
15111511

0 commit comments

Comments
 (0)