From 188c194b0f7bad1aa6c5db46e04c3ef63d2b10e6 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 2 Jul 2022 15:17:28 -0700 Subject: [PATCH] Left some experiment stem code in convnext by mistake --- timm/models/convnext.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index 662695c7..138e5030 100644 --- 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