Merge pull request #979 from cuicheng01/develop

update new_user/ShuffleNetV2_x0_25.yaml
2025-06-03 21:55:06 +08:00 · 2021-06-29 11:05:23 +08:00 · 2021-06-29 11:05:23 +08:00 · 6fc9e5ad54
commit 6fc9e5ad54
parent c43b40aab3 8a31e4f4d7
4 changed files with 2 additions and 6 deletions
--- 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())

--- 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]
--- 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]
--- 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