Show related heads in single view

catherinesyeh / attention-viz

Visualizing query-key interactions in language + vision transformers

http://attentionviz.com/

MIT License

127 stars 17 forks source link

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?