diff --git a/ppcls/arch/backbone/model_zoo/xception_deeplab.py b/ppcls/arch/backbone/model_zoo/xception_deeplab.py index f2424823f..dc8c234ec 100644 --- a/ppcls/arch/backbone/model_zoo/xception_deeplab.py +++ b/ppcls/arch/backbone/model_zoo/xception_deeplab.py @@ -22,7 +22,7 @@ from paddle.nn import AdaptiveAvgPool2D, MaxPool2D, AvgPool2D from ppcls.utils.save_load import load_dygraph_pretrain, load_dygraph_pretrain_from_url MODEL_URLS = {"Xception41_deeplab": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/Xception41_deeplab_pretrained.pdparams", - "Xception65_deeplab": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/Xception41_deeplab_pretrained.pdparams"} + "Xception65_deeplab": "https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/Xception65_deeplab_pretrained.pdparams"} __all__ = list(MODEL_URLS.keys()) diff --git a/ppcls/configs/ImageNet/Xception/Xception65.yaml b/ppcls/configs/ImageNet/Xception/Xception65.yaml index 5b772e6fe..b875e40ec 100644 --- a/ppcls/configs/ImageNet/Xception/Xception65.yaml +++ b/ppcls/configs/ImageNet/Xception/Xception65.yaml @@ -125,8 +125,6 @@ Infer: Metric: Train: - - TopkAcc: - topk: [1, 5] Eval: - TopkAcc: topk: [1, 5] diff --git a/ppcls/configs/ImageNet/Xception/Xception71.yaml b/ppcls/configs/ImageNet/Xception/Xception71.yaml index ac244604f..3272cc9cf 100644 --- a/ppcls/configs/ImageNet/Xception/Xception71.yaml +++ b/ppcls/configs/ImageNet/Xception/Xception71.yaml @@ -125,8 +125,6 @@ Infer: Metric: Train: - - TopkAcc: - topk: [1, 5] Eval: - TopkAcc: topk: [1, 5] diff --git a/ppcls/configs/quick_start/new_user/ShuffleNetV2_x0_25.yaml b/ppcls/configs/quick_start/new_user/ShuffleNetV2_x0_25.yaml index 013cf0c36..05739ac85 100644 --- a/ppcls/configs/quick_start/new_user/ShuffleNetV2_x0_25.yaml +++ b/ppcls/configs/quick_start/new_user/ShuffleNetV2_x0_25.yaml @@ -3,7 +3,7 @@ Global: checkpoints: null pretrained_model: null output_dir: ./output/ - device: gpu + device: cpu class_num: 102 save_interval: 1 eval_during_train: True