mochan-b/Transformer-Attention-Maps

Exploration of transformer attention maps

Python

Transformer-Attention-Maps

Exploration of transformer attention maps

Heavily based on the tutorial notebook at: https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/tutorial6/Transformers_and_MHAttention.html