Merge pull request #1454 from cuicheng01/release/2.2

[cherry-pick]fix class_num in quickstart
pull/2482/head
cuicheng01 2021-11-22 15:52:02 +08:00 committed by GitHub
commit e360fd6b71
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions

View File

@ -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: