PROJECT
Vision Transformers PyTorch Toy Implementation
A from-scratch PyTorch implementation of the original Vision Transformer for image classification. It breaks the architecture into patch embedding, transformer encoder and classification-head components.