mashaan14 / VisionTransformer-MNIST

This notebook is designed to plot the attention maps of a vision transformer trained on MNIST digits.
27 stars 8 forks source link