change config to baseline
@@ -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
