Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
LTSLegendary

lucidrains/vit-pytorch

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

Owner
lucidrains (personal account)
Stars
Stars: 25,555
Forks
Forks: 3,494
Language
Python
License
MIT
Created
2020
Topics
artificial-intelligenceattention-mechanismtransformerscomputer-visionimage-classification
Tier
LTS (20,000 ★ and up) What this tier means
View on GitHub

More repos by lucidrains

All 175 repos by lucidrains

Nearby in Python

All Python repos