diff --git a/ppcls/arch/backbone/legendary_models/pp_hgnet.py b/ppcls/arch/backbone/legendary_models/pp_hgnet.py index 3e0412dfb210c7dc44bc98854dbb96fca526ab1f..5d86a15932b8238353eeca7ba038c30551c935d5 100644 --- a/ppcls/arch/backbone/legendary_models/pp_hgnet.py +++ b/ppcls/arch/backbone/legendary_models/pp_hgnet.py @@ -27,7 +27,8 @@ MODEL_URLS = { "PPHGNet_tiny": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPHGNet_tiny_pretrained.pdparams", "PPHGNet_small": - "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPHGNet_small_pretrained.pdparams" + "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/legendary_models/PPHGNet_small_pretrained.pdparams", + "PPHGNet_base": "" } __all__ = list(MODEL_URLS.keys()) diff --git a/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml b/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml index a26ec48e7678bd30443de3b0ab9524f2c3cb5af4..7af9680cb2ec90c4c70cdf66d1c7f5a6225a9456 100644 --- a/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml +++ b/ppcls/configs/ImageNet/Distillation/res2net200_vd_distill_pphgnet_base.yaml @@ -54,9 +54,8 @@ Loss: model_name_pairs: - ["Student", "Teacher"] Eval: - - DistillationGTCELoss: + - CELoss: weight: 1.0 - model_names: ["Student"] Optimizer: