Hrazhan/vit

Vision Transformer (ViT) model for image classification using PyTorch

Vision Transformer (ViT) Replication

This repository contains code for replicating the Vision Transformer (ViT) model proposed in the paper "An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale" by Dosovitskiy et al.

References

• Official implementation of ViT
• ViT paper by Dosovitskiy et al.