提交 2ad0ca44 编写于 作者: W WenmuZhou

update db config

上级 d3b609ee
......@@ -2,7 +2,7 @@ Global:
use_gpu: true
epoch_num: 1200
log_smooth_window: 20
print_batch_step: 2
print_batch_step: 10
save_model_dir: ./output/db_mv3/
save_epoch_step: 1200
# evaluation is run every 5000 iterations after the 4000th iteration
......@@ -39,7 +39,7 @@ Loss:
alpha: 5
beta: 10
ohem_ratio: 3
Optimizer:
name: Adam
beta1: 0.9
......@@ -100,7 +100,7 @@ Train:
loader:
shuffle: True
drop_last: False
batch_size_per_card: 4
batch_size_per_card: 16
num_workers: 8
Eval:
......@@ -128,4 +128,4 @@ Eval:
shuffle: False
drop_last: False
batch_size_per_card: 1 # must be 1
num_workers: 2
\ No newline at end of file
num_workers: 8
\ No newline at end of file
......@@ -6,7 +6,7 @@ Global:
save_model_dir: ./output/det_r50_vd/
save_epoch_step: 1200
# evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step: [5000,4000]
eval_batch_step: [4000,5000]
# if pretrained_model is saved in static mode, load_static_weights must set to True
load_static_weights: True
cal_metric_during_train: False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册