From dd08d344afe8e634b357da1d3684abd4e524624d Mon Sep 17 00:00:00 2001
From: cuicheng01 <cuicheng_smile@163.com>
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 d18f8dcb..89cafca3 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