s-chh / PyTorch-Scratch-Vision-Transformer-ViT

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.
129Updated last month

Alternatives and similar repositories for PyTorch-Scratch-Vision-Transformer-ViT:

Users that are interested in PyTorch-Scratch-Vision-Transformer-ViT are comparing it to the libraries listed below