catherinesyeh / attention-viz

Visualizing query-key interactions in language + vision transformers
http://attentionviz.com/
MIT License
122 stars 15 forks source link

Show related heads in single view #52

Open catherinesyeh opened 1 year ago

catherinesyeh commented 1 year ago

Maybe show 2-3 other attention heads with similar attention patterns in the corner or something?