mashaan14 / VisionTransformer-MNIST

This notebook is designed to plot the attention maps of a vision transformer trained on MNIST digits.
30Updated last month

Related projects

Alternatives and complementary repositories for VisionTransformer-MNIST