how to get confidence score of each word using Attn
nbhupendra opened this issue · 2 comments
nbhupendra commented
nbhupendra commented
ku21fan commented
Hello,
Sorry for the late reply.
I just commit codes about confidence scores.
Please check and test recent version~ (test.py and demo.py) e8b465d
P.S. Recent code may occur error in F.grid_sample(~~, align_corners=True)
because the attribute align_corners
of F.grid_sample is introduced in PyTorch 1.3.0.
So you may need to remove , align_corners=True
in here or upgrade PyTorch to 1.3.0
Best