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

change config to baseline

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