rename `ConvNext` to `ConvNeXt`
parent
436f64ad84
commit
4266b8790d
ppcls
arch/backbone
model_zoo
configs/ImageNet/ConvNeXt
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue