|
|
|
@ -29,9 +29,7 @@ Optimizer:
|
|
|
|
|
beta1: 0.9
|
|
|
|
|
beta2: 0.999
|
|
|
|
|
learning_rate:
|
|
|
|
|
name: Cosine
|
|
|
|
|
lr: 0.001
|
|
|
|
|
warmup_epoch: 4
|
|
|
|
|
regularizer:
|
|
|
|
|
name: 'L2'
|
|
|
|
|
factor: 0.00001
|
|
|
|
@ -79,7 +77,7 @@ TRAIN:
|
|
|
|
|
- RecResizeImg:
|
|
|
|
|
image_shape: [ 3,32,320 ]
|
|
|
|
|
- keepKeys:
|
|
|
|
|
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
|
|
|
|
|
keep_keys: [ 'image','label','length' ] # dataloader will return list in this order
|
|
|
|
|
loader:
|
|
|
|
|
batch_size: 256
|
|
|
|
|
shuffle: True
|
|
|
|
@ -100,7 +98,7 @@ EVAL:
|
|
|
|
|
- RecResizeImg:
|
|
|
|
|
image_shape: [ 3,32,320 ]
|
|
|
|
|
- keepKeys:
|
|
|
|
|
keep_keys: [ 'image','label','length' ] # dataloader将按照此顺序返回list
|
|
|
|
|
keep_keys: [ 'image','label','length' ] # dataloader will return list in this order
|
|
|
|
|
loader:
|
|
|
|
|
shuffle: False
|
|
|
|
|
drop_last: False
|
|
|
|
|