Merge pull request #1454 from cuicheng01/release/2.2
[cherry-pick]fix class_num in quickstartpull/2482/head
commit
e360fd6b71
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue