diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py
index 8c260a92a2d60b4896a1e115db25493670e85fad..aebd97fba15282dd8961d30ce6cc9e79a26f45ca 100644
--- a/ppocr/data/imaug/label_ops.py
+++ b/ppocr/data/imaug/label_ops.py
@@ -318,7 +318,7 @@ class AttnLabelEncode(BaseRecLabelEncode):
         text = self.encode(text)
         if text is None:
             return None
-        if len(text) >= self.max_text_len:
+        if len(text) >= self.max_text_len - 1:
             return None
         data['length'] = np.array(len(text))
         text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len
diff --git a/ppocr/modeling/heads/rec_att_head.py b/ppocr/modeling/heads/rec_att_head.py
index 6d77e42eb5def579052687ab6fdc265159311884..3c572af88e9c7eef1e313eb09d467b5110156f8e 100644
--- a/ppocr/modeling/heads/rec_att_head.py
+++ b/ppocr/modeling/heads/rec_att_head.py
@@ -75,6 +75,7 @@ class AttentionHead(nn.Layer):
                             probs_step, axis=1)], axis=1)
                 next_input = probs_step.argmax(axis=1)
                 targets = next_input
+        if not self.training:
             probs = paddle.nn.functional.softmax(probs, axis=2)
         return probs