From 8f0bc0591e8fe4803cae892f860d75401b517d6c Mon Sep 17 00:00:00 2001 From: SeeFun Date: Tue, 5 Apr 2022 20:00:57 +0800 Subject: [PATCH] fix convnext args --- timm/models/convnext.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/timm/models/convnext.py b/timm/models/convnext.py index 9fd4525a..1aacef2b 100644 --- a/timm/models/convnext.py +++ b/timm/models/convnext.py @@ -421,14 +421,14 @@ def convnext_large(pretrained=False, **kwargs): @register_model def convnext_tiny_in22ft1k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=(3, 3, 9, 3), dims=(96, 192, 384, 768), **kwargs) model = _create_convnext('convnext_tiny_in22ft1k', pretrained=pretrained, **model_args) return model @register_model def convnext_small_in22ft1k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=[3, 3, 27, 3], dims=[96, 192, 384, 768], **kwargs) model = _create_convnext('convnext_small_in22ft1k', pretrained=pretrained, **model_args) return model @@ -456,14 +456,14 @@ def convnext_xlarge_in22ft1k(pretrained=False, **kwargs): @register_model def convnext_tiny_384_in22ft1k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=(3, 3, 9, 3), dims=(96, 192, 384, 768), **kwargs) model = _create_convnext('convnext_tiny_384_in22ft1k', pretrained=pretrained, **model_args) return model @register_model def convnext_small_384_in22ft1k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=[3, 3, 27, 3], dims=[96, 192, 384, 768], **kwargs) model = _create_convnext('convnext_small_384_in22ft1k', pretrained=pretrained, **model_args) return model @@ -491,14 +491,14 @@ def convnext_xlarge_384_in22ft1k(pretrained=False, **kwargs): @register_model def convnext_tiny_in22k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=(3, 3, 9, 3), dims=(96, 192, 384, 768), **kwargs) model = _create_convnext('convnext_tiny_in22k', pretrained=pretrained, **model_args) return model @register_model def convnext_small_in22k(pretrained=False, **kwargs): - model_args = dict(depths=[3, 3, 27, 3], dims=[128, 256, 512, 1024], **kwargs) + model_args = dict(depths=[3, 3, 27, 3], dims=[96, 192, 384, 768], **kwargs) model = _create_convnext('convnext_small_in22k', pretrained=pretrained, **model_args) return model