diff --git a/ppcls/configs/quick_start/ResNet50_vd.yaml b/ppcls/configs/quick_start/ResNet50_vd.yaml index 1b4e844fa7fc9cfe3cfa1b8aec1d4379c507cf2b..0a3258b2afe9d12ba80d55975057b4e272b353e4 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: