提交 dd08d344 编写于 作者: C cuicheng01

fix class_num in quickstart

上级 9a23b432
...@@ -4,7 +4,6 @@ Global: ...@@ -4,7 +4,6 @@ Global:
pretrained_model: null pretrained_model: null
output_dir: ./output/ output_dir: ./output/
device: gpu device: gpu
class_num: 102
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -17,8 +16,9 @@ Global: ...@@ -17,8 +16,9 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ResNet50_vd name: ResNet50_vd
class_num: 102
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
Train: Train:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册