fix class_num in quickstart

2021-11-17 15:04:04 +00:00 · 2021-11-17 15:04:04 +00:00 · dd08d344af
parent 9a23b4326a
commit dd08d344af
1 changed files with 3 additions and 3 deletions
--- a/ppcls/configs/quick_start/ResNet50_vd.yaml
+++ b/ppcls/configs/quick_start/ResNet50_vd.yaml
@ -4,7 +4,6 @@ Global:
  pretrained_model: null
  output_dir: ./output/
  device: gpu
-  class_num: 102
  save_interval: 1
  eval_during_train: True
  eval_interval: 1
@ -17,8 +16,9 @@ Global:

 # model architecture
 Arch:
-  name: ResNet50_vd
- 
+  name: ResNet50_vd 
+  class_num: 102
+
 # loss function config for traing/eval process
 Loss:
  Train: