fix attn score

2ce4d9f8 · tink2123 · 5aa14c5f · 2ce4d9f8
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

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

未找到文件。
--- 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