Attention Arrow Visualization for VIT-16/32

catherinesyeh / attention-viz

Visualizing query-key interactions in language + vision transformers

http://attentionviz.com/

MIT License

127 stars 17 forks source link

Closed yc015 closed 1 year ago

yc015 commented 1 year ago

[x] In the side panel, add an visualization that draw arrows between the patch that paid most attention to each other.

yc015 commented 1 year ago

Done!