提交 0d22017e 编写于 作者: T Tingquan Gao 提交者: Tingquan Gao

fix: move class_num from Global to Arch

上级 bb6581d2
...@@ -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: 1000
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -17,6 +16,7 @@ Global: ...@@ -17,6 +16,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ESNet_x0_25 name: ESNet_x0_25
class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -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: 1000
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -17,6 +16,7 @@ Global: ...@@ -17,6 +16,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ESNet_x0_5 name: ESNet_x0_5
class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -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: 1000
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -17,6 +16,7 @@ Global: ...@@ -17,6 +16,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ESNet_x0_75 name: ESNet_x0_75
class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
...@@ -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: 1000
save_interval: 1 save_interval: 1
eval_during_train: True eval_during_train: True
eval_interval: 1 eval_interval: 1
...@@ -17,6 +16,7 @@ Global: ...@@ -17,6 +16,7 @@ Global:
# model architecture # model architecture
Arch: Arch:
name: ESNet_x1_0 name: ESNet_x1_0
class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册