/CLIP-self-attention-visualization

Plotting heatmaps with the self-attention of the [CLS] tokens in the last layer.

Primary LanguagePython

No issues in this repository yet.