/ViT

Visual Transformer implementation in PyTorch.

Primary LanguageJupyter Notebook

Visual Transformer Implementation

After reading The Illustrated Transformer blog post, I needed to implement one myself in order to gain a better understanding of the model's architecture. I replicated the code from another PyTorch implementation of ViT here.