Seeking methods to visualize attention

[xiaoc57]

Hello. I noticed that your demo shows attention visualization. However, I'm not sure how to do it. I'd appreciate any help.

[rover-xingyu]

Hi @xiaoc57, you can return the learning rate and visualize latter.