From ab9c3fe9d55a40da99b0df64bb02e90e48341470 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Wed, 17 Nov 2021 15:04:04 +0000 Subject: [PATCH] fix class_num in quickstart --- ppcls/configs/quick_start/ResNet50_vd.yaml | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/ppcls/configs/quick_start/ResNet50_vd.yaml b/ppcls/configs/quick_start/ResNet50_vd.yaml index 1b4e844f..0a3258b2 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: -- GitLab