Simple and easy to understand PyTorch implementation of Vision Transformer (ViT) from scratch, with detailed steps. Tested on common datasets like MNIST, CIFAR10, and more.
-
Updated
Dec 23, 2024 - Python
Simple and easy to understand PyTorch implementation of Vision Transformer (ViT) from scratch, with detailed steps. Tested on common datasets like MNIST, CIFAR10, and more.
PyTorch implementation of ViT from scratch
Minimalist MNIST implementation with two hidden layers written in C
Add a description, image, and links to the vit-mnist topic page so that developers can more easily learn about it.
To associate your repository with the vit-mnist topic, visit your repo's landing page and select "manage topics."