diff --git a/ppcls/arch/backbone/__init__.py b/ppcls/arch/backbone/__init__.py index 49d47bb7e522ca782087807d31e736e087189441..669547c8aa58f65caa6b74b55b5b7c55570a9a8e 100644 --- a/ppcls/arch/backbone/__init__.py +++ b/ppcls/arch/backbone/__init__.py @@ -66,7 +66,7 @@ from .model_zoo.cspnet import CSPDarkNet53 from .model_zoo.pvt_v2 import PVT_V2_B0, PVT_V2_B1, PVT_V2_B2_Linear, PVT_V2_B2, PVT_V2_B3, PVT_V2_B4, PVT_V2_B5 from .model_zoo.mobilevit import MobileViT_XXS, MobileViT_XS, MobileViT_S from .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 .model_zoo.van import VAN_tiny +from .model_zoo.van import VAN_B0 from .model_zoo.peleenet import PeleeNet from .model_zoo.convnext import ConvNeXt_tiny from .model_zoo.cae import cae_base_patch16_224, cae_large_patch16_224 diff --git a/ppcls/arch/backbone/model_zoo/van.py b/ppcls/arch/backbone/model_zoo/van.py index c4d7339292faf8b11c9a4a637b7ae357b1bae1fd..febf45113b00d24b494180fee244ee82a1d72e30 100644 --- a/ppcls/arch/backbone/model_zoo/van.py +++ b/ppcls/arch/backbone/model_zoo/van.py @@ -24,7 +24,7 @@ from paddle.nn.initializer import TruncatedNormal, Constant from ....utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url MODEL_URLS = { - "VAN_tiny": "", # TODO + "VAN_B0": "", # TODO } __all__ = list(MODEL_URLS.keys()) @@ -306,7 +306,7 @@ def _load_pretrained(pretrained, model, model_url, use_ssld=False): ) -def VAN_tiny(pretrained=False, use_ssld=False, **kwargs): +def VAN_B0(pretrained=False, use_ssld=False, **kwargs): model = VAN(embed_dims=[32, 64, 160, 256], mlp_ratios=[8, 8, 4, 4], norm_layer=partial( @@ -314,5 +314,5 @@ def VAN_tiny(pretrained=False, use_ssld=False, **kwargs): depths=[3, 3, 5, 2], **kwargs) _load_pretrained( - pretrained, model, MODEL_URLS["VAN_tiny"], use_ssld=use_ssld) + pretrained, model, MODEL_URLS["VAN_B0"], use_ssld=use_ssld) return model diff --git a/ppcls/configs/ImageNet/VAN/VAN_tiny.yaml b/ppcls/configs/ImageNet/VAN/VAN_B0.yaml similarity index 99% rename from ppcls/configs/ImageNet/VAN/VAN_tiny.yaml rename to ppcls/configs/ImageNet/VAN/VAN_B0.yaml index ae08dcfa9e99d0ccab4f051abe9351ff61c9b736..d72f0faea3bd4f65e0c23b1a74e9075bd2e7d92c 100644 --- a/ppcls/configs/ImageNet/VAN/VAN_tiny.yaml +++ b/ppcls/configs/ImageNet/VAN/VAN_B0.yaml @@ -18,7 +18,7 @@ Global: # model architecture Arch: - name: VAN_tiny + name: VAN_B0 class_num: 1000 drop_path_rate: 0.1 drop_rate: 0.0