Merge pull request #1453 from cuicheng01/develop

fix class_num in quickstart
2021-11-22 12:00:56 +08:00 · 2021-11-22 12:00:56 +08:00 · bc16eed9f9
parent 6f06b677e6 dd08d344af
commit bc16eed9f9
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
@ -18,6 +17,7 @@ Global:
 # model architecture
 Arch:
  name: ResNet50_vd 
+  class_num: 102

 # loss function config for traing/eval process
 Loss: