mashaan14 / VisionTransformer-MNIST

This notebook is designed to plot the attention maps of a vision transformer trained on MNIST digits.
27Updated 3 weeks ago

Related projects: