Left some experiment stem code in convnext by mistake

3 years ago · 188c194b0f
parent 70d6d2c484
commit 188c194b0f
1 changed files with 1 additions and 1 deletions
--- a/timm/models/convnext.py
+++ b/timm/models/convnext.py
@ -434,7 +434,7 @@ def convnext_tiny_hnf(pretrained=False, **kwargs):
@register_model
 def convnext_tiny_hnfd(pretrained=False, **kwargs):
    model_args = dict(
-        depths=(3, 3, 9, 3), dims=(96, 192, 384, 768), head_norm_first=True, conv_mlp=True, stem_type='dual', **kwargs)
+        depths=(3, 3, 9, 3), dims=(96, 192, 384, 768), head_norm_first=True, conv_mlp=True, **kwargs)
    model = _create_convnext('convnext_tiny_hnf', pretrained=pretrained, **model_args)
    return model