diff --git a/ppcls/configs/quick_start/ResNet50_vd.yaml b/ppcls/configs/quick_start/ResNet50_vd.yaml index d18f8dcb7aa019198f36bff8e47d9a12674a4caa..89cafca3817753453c1a0de1571bc1eb5e532577 100644 --- 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: