Merge pull request PaddlePaddle#4557 from tink2123/cherry-pick

[cherry-pick] fix attn score
Qs-Tim · Nov 5, 2021 · bb6d66a · bb6d66a
2 parents 626704d + 576644b
commit bb6d66a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ppocr/modeling/heads/rec_att_head.py b/ppocr/modeling/heads/rec_att_head.py
@@ -75,7 +75,7 @@ def forward(self, inputs, targets=None, batch_max_length=25):
                             probs_step, axis=1)], axis=1)
                 next_input = probs_step.argmax(axis=1)
                 targets = next_input
-
+            probs = paddle.nn.functional.softmax(probs, axis=2)
         return probs