Skip to content

Commit c6e5557

Browse files
committed
Mismatch pretrained_cfg
1 parent 5d7bd29 commit c6e5557

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

timm/models/convnext.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -996,7 +996,7 @@ def convnext_zepto_rms_ols(pretrained=False, **kwargs) -> ConvNeXt:
996996
# timm femto variant (NOTE: still tweaking depths, will vary between 3-4M param, current is 3.7M
997997
model_args = dict(
998998
depths=(2, 2, 4, 2), dims=(32, 64, 128, 256), conv_mlp=True, norm_layer='rmsnorm2d', stem_type='overlap_act')
999-
model = _create_convnext('convnext_zepto_rms_oas', pretrained=pretrained, **dict(model_args, **kwargs))
999+
model = _create_convnext('convnext_zepto_rms_ols', pretrained=pretrained, **dict(model_args, **kwargs))
10001000
return model
10011001

10021002

0 commit comments

Comments
 (0)