s-chh / pytorch-scratch-vision-transformer-vit Goto Github PK
View Code? Open in Web Editor NEWSimple, scratch and easy to understand PyTorch implementation of Vision Transformer (ViT) for small datasets like MNIST, FashionMNIST, SVHN and CIFAR10 with detailed steps.