Skip to content

GitLab

Commit 9c59ae1c authored by tink2123's avatar tink2123
Browse files

fix bug

parent 349c51e6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/modeling/architectures/rec_model.py
View file @ 9c59ae1c
...@@ -58,7 +58,7 @@ class RecModel(object): ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment