提交 4266b879 编写于 作者: Y Yang Nie

rename `ConvNext` to `ConvNeXt`

上级 436f64ad
...@@ -67,7 +67,7 @@ from ppcls.arch.backbone.model_zoo.pvt_v2 import PVT_V2_B0, PVT_V2_B1, PVT_V2_B2 ...@@ -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.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.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.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.resnet_variant import ResNet50_last_stage_stride1
from ppcls.arch.backbone.variant_models.vgg_variant import VGG19Sigmoid from ppcls.arch.backbone.variant_models.vgg_variant import VGG19Sigmoid
from ppcls.arch.backbone.variant_models.pp_lcnet_variant import PPLCNet_x2_5_Tanh from ppcls.arch.backbone.variant_models.pp_lcnet_variant import PPLCNet_x2_5_Tanh
......
...@@ -29,7 +29,7 @@ from paddle.nn.initializer import TruncatedNormal, Constant ...@@ -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 from ppcls.utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url
MODEL_URLS = { MODEL_URLS = {
"ConvNext_tiny": "", # TODO "ConvNeXt_tiny": "", # TODO
} }
__all__ = list(MODEL_URLS.keys()) __all__ = list(MODEL_URLS.keys())
...@@ -233,8 +233,8 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=False): ...@@ -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) model = ConvNeXt(depths=[3, 3, 9, 3], dims=[96, 192, 384, 768], **kwargs)
_load_pretrained( _load_pretrained(
pretrained, model, MODEL_URLS["ConvNext_tiny"], use_ssld=use_ssld) pretrained, model, MODEL_URLS["ConvNeXt_tiny"], use_ssld=use_ssld)
return model return model
...@@ -24,7 +24,7 @@ EMA: ...@@ -24,7 +24,7 @@ EMA:
# model architecture # model architecture
Arch: Arch:
name: ConvNext_tiny name: ConvNeXt_tiny
class_num: 1000 class_num: 1000
drop_path_rate: 0.1 drop_path_rate: 0.1
layer_scale_init_value: 1e-6 layer_scale_init_value: 1e-6
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册