This commit is contained in:
tink2123 2020-06-05 18:00:52 +08:00
parent 4cac91eb7b
commit aab46214fd
3 changed files with 3 additions and 1 deletions

View File

@ -13,6 +13,7 @@ Global:
max_text_length: 25
character_type: en
loss_type: attention
tps: true
reader_yml: ./configs/rec/rec_benchmark_reader.yml
pretrain_weights:
checkpoints:

View File

@ -13,6 +13,7 @@ Global:
max_text_length: 25
character_type: en
loss_type: ctc
tps: true
reader_yml: ./configs/rec/rec_benchmark_reader.yml
pretrain_weights:
checkpoints:

View File

@ -184,7 +184,7 @@ class SimpleReader(object):
self.infer_img = params['infer_img']
self.use_tps = False
if "tps" in params:
self.ues_tps = True
self.use_tps = True
if params['mode'] == 'train':
self.batch_size = params['train_batch_size_per_card']
self.drop_last = True