From 4266b8790d018a9e8146e5e8da83568a887528d8 Mon Sep 17 00:00:00 2001 From: Yang Nie Date: Sat, 9 Jul 2022 23:25:50 +0800 Subject: [PATCH] rename `ConvNext` to `ConvNeXt` --- ppcls/arch/backbone/__init__.py | 2 +- ppcls/arch/backbone/model_zoo/convnext.py | 6 +++--- ppcls/configs/ImageNet/ConvNeXt/ConvNeXt_tiny.yaml | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/ppcls/arch/backbone/__init__.py b/ppcls/arch/backbone/__init__.py index 4c7dc908..efea4ec4 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 9e3e7e78..f30894ea 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 0ce3b57f..fb6e3cbd 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 -- GitLab