fix attn score

2ce4d9f8 · tink2123 · 5aa14c5f · 2ce4d9f8
Commit 2ce4d9f8 authored Nov 04, 2021 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/modeling/heads/rec_att_head.py ppocr/modeling/heads/rec_att_head.py +1 -1

No files found.
--- a/ppocr/modeling/heads/rec_att_head.py
+++ b/ppocr/modeling/heads/rec_att_head.py
@@ -75,7 +75,7 @@ class AttentionHead(nn.Layer):
                            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