diff --git a/ppocr/modeling/architectures/rec_model.py b/ppocr/modeling/architectures/rec_model.py
index 91f778ce97768bc8079affb0e974d10da6c195d4..fe2d4c16dce3882980fe2238ecc16c7c08a89792 100755
--- a/ppocr/modeling/architectures/rec_model.py
+++ b/ppocr/modeling/architectures/rec_model.py
@@ -58,7 +58,7 @@ class RecModel(object):
         self.loss_type = global_params['loss_type']
         self.image_shape = global_params['image_shape']
         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"]
         else:
             self.num_heads = None