diff --git a/ppcls/arch/backbone/__init__.py b/ppcls/arch/backbone/__init__.py index 4c7dc90860d0d3821fbb36f8f65616c1b0a6051d..efea4ec46860e012d58274ac911a8144dfdef0e2 100644 --- a/ppcls/arch/backbone/__init__.py +++ b/ppcls/arch/backbone/__init__.py @@ -67,7 +67,7 @@ from ppcls.arch.backbone.model_zoo.pvt_v2 import PVT_V2_B0, PVT_V2_B1, PVT_V2_B2 from ppcls.arch.backbone.model_zoo.mobilevit import MobileViT_XXS, MobileViT_XS, MobileViT_S from ppcls.arch.backbone.model_zoo.repvgg import RepVGG_A0, RepVGG_A1, RepVGG_A2, RepVGG_B0, RepVGG_B1, RepVGG_B2, RepVGG_B1g2, RepVGG_B1g4, RepVGG_B2g4, RepVGG_B3g4 from ppcls.arch.backbone.model_zoo.van import VAN_tiny -from ppcls.arch.backbone.model_zoo.convnext import ConvNext_tiny +from ppcls.arch.backbone.model_zoo.convnext import ConvNeXt_tiny from ppcls.arch.backbone.variant_models.resnet_variant import ResNet50_last_stage_stride1 from ppcls.arch.backbone.variant_models.vgg_variant import VGG19Sigmoid from ppcls.arch.backbone.variant_models.pp_lcnet_variant import PPLCNet_x2_5_Tanh diff --git a/ppcls/arch/backbone/model_zoo/convnext.py b/ppcls/arch/backbone/model_zoo/convnext.py index 9e3e7e78790c1777d3eea0be0753cd2fb4df6e87..f30894eab526b8deb5e61a964dc287415f1b1a02 100644 --- a/ppcls/arch/backbone/model_zoo/convnext.py +++ b/ppcls/arch/backbone/model_zoo/convnext.py @@ -29,7 +29,7 @@ from paddle.nn.initializer import TruncatedNormal, Constant from ppcls.utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url MODEL_URLS = { - "ConvNext_tiny": "", # TODO + "ConvNeXt_tiny": "", # TODO } __all__ = list(MODEL_URLS.keys()) @@ -233,8 +233,8 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=False): ) -def ConvNext_tiny(pretrained=False, use_ssld=False, **kwargs): +def ConvNeXt_tiny(pretrained=False, use_ssld=False, **kwargs): model = ConvNeXt(depths=[3, 3, 9, 3], dims=[96, 192, 384, 768], **kwargs) _load_pretrained( - pretrained, model, MODEL_URLS["ConvNext_tiny"], use_ssld=use_ssld) + pretrained, model, MODEL_URLS["ConvNeXt_tiny"], use_ssld=use_ssld) return model diff --git a/ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml b/ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml index 0ce3b57fee198dddf03514890948964ef973a96c..fb6e3cbdbb2dc648e4ef0bd1cad59106efbf91db 100644 --- a/ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml +++ b/ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml @@ -24,7 +24,7 @@ EMA: # model architecture Arch: - name: ConvNext_tiny + name: ConvNeXt_tiny class_num: 1000 drop_path_rate: 0.1 layer_scale_init_value: 1e-6