Display CLS token for VIT-16/32 - Githubissues

catherinesyeh / attention-viz

Visualizing query-key interactions in language + vision transformers

http://attentionviz.com/

MIT License

122 stars 15 forks source link

Display CLS token for VIT-16/32 #61

Closed yc015 closed 1 year ago

yc015 commented 1 year ago

[x] Display CLS token for VIT-16/32 visualization
[x] Also include CLS token in side-panel attention heatmap

yc015 commented 1 year ago

Done!

1678666929385

1678667083108