提交 9ac9cb8a 编写于 作者: S sibo2rr

add config file

上级 f2dde176
...@@ -15,12 +15,11 @@ Global: ...@@ -15,12 +15,11 @@ Global:
save_inference_dir: ./inference save_inference_dir: ./inference
# training model under @to_static # training model under @to_static
to_static: False to_static: False
use_dali: True
# model architecture # model architecture
Arch: Arch:
name: MobileNetV1 name: MobileNetV1
class_num: 100 class_num: 1000
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
...@@ -68,8 +67,8 @@ DataLoader: ...@@ -68,8 +67,8 @@ DataLoader:
sampler: sampler:
name: MultiScaleSamplerDDP name: MultiScaleSamplerDDP
scales: [224, 256] scales: [160, 192, 256, 288, 320]
first_bs: 4 first_bs: 64
is_training: True is_training: True
loader: loader:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册