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