castorini/daam

Support for self attention visualization

thakral-kartik opened this issue · 1 comments

thakral-kartik commented

Hi, I find your work very interesting. Is there a way to visualize self attention maps also in the existing repo?

RishiDarkDevil commented

Hi @thakral-kartik, you can visit my forked-repo for doing that: https://github.com/RishiDarkDevil/daam-i2i