yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-12-28 07:28:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix attn score

Browse Source
This commit is contained in:
tink2123 2021-11-04 15:58:00 +00:00
parent 5aa14c5f11
commit 2ce4d9f8b5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ppocr/modeling/heads/rec_att_head.py
View File

@ -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