forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix bug

This commit is contained in:
tink2123 2020-08-17 01:43:59 +08:00
parent 349c51e6bf
commit 9c59ae1cd5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/modeling/architectures/rec_model.py
View File

@ -58,7 +58,7 @@ class RecModel(object):
self.loss_type = global_params['loss_type'] self.loss_type = global_params['loss_type']
self.image_shape = global_params['image_shape'] self.image_shape = global_params['image_shape']
self.max_text_length = global_params['max_text_length'] self.max_text_length = global_params['max_text_length']
if "num_heads" in params: if "num_heads" in global_params:
self.num_heads = global_params["num_heads"] self.num_heads = global_params["num_heads"]
else: else:
self.num_heads = None self.num_heads = None