提交 0c183d25 编写于 作者: W WenmuZhou

change config to baseline

上级 cafb2df8
......@@ -6,7 +6,7 @@ Global:
save_model_dir: ./output/rec/mv3_none_bilstm_ctc/
save_epoch_step: 3
# evaluation is run every 5000 iterations after the 4000th iteration
eval_batch_step: [0, 1000]
eval_batch_step: [0, 2000]
# if pretrained_model is saved in static mode, load_static_weights must set to True
cal_metric_during_train: True
pretrained_model:
......@@ -18,11 +18,8 @@ Global:
character_dict_path:
character_type: en
max_text_length: 25
loss_type: ctc
infer_mode: False
# use_space_char: True
# use_tps: False
use_space_char: False
Optimizer:
......@@ -33,7 +30,7 @@ Optimizer:
learning_rate: 0.0005
regularizer:
name: 'L2'
factor: 0.00001
factor: 0
Architecture:
model_type: rec
......@@ -49,7 +46,7 @@ Architecture:
hidden_size: 96
Head:
name: CTCHead
fc_decay: 0.0004
fc_decay: 0
Loss:
name: CTCLoss
......@@ -75,8 +72,8 @@ Train:
- KeepKeys:
keep_keys: ['image', 'label', 'length'] # dataloader will return list in this order
loader:
shuffle: True
batch_size_per_card: 256
shuffle: False
drop_last: True
num_workers: 8
......@@ -97,4 +94,4 @@ Eval:
shuffle: False
drop_last: False
batch_size_per_card: 256
num_workers: 2
num_workers: 4
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册