vit-pytorch 0.40.2


Vision Transformer (ViT) - Pytorch

Vision Transformer (ViT) - Pytorch

Stars: 12532, Watchers: 12532, Forks: 2077, Open Issues: 100

The lucidrains/vit-pytorch repo was created 2 years ago and was last updated 3 hours ago.
The project is extremely popular with a mindblowing 12532 github stars!

How to Install vit-pytorch

You can install vit-pytorch using pip

pip install vit-pytorch

or add it to a project with poetry

poetry add vit-pytorch

Package Details

Phil Wang
GitHub Repo


  • Scientific/Engineering/Artificial Intelligence
No  vit-pytorch  pypi packages just yet.


A list of common vit-pytorch errors.

Code Examples

Here are some vit-pytorch code examples and snippets.

GitHub Issues

The vit-pytorch package has 100 open issues on GitHub

  • MAE using pretrained VIT
  • How to get the feature map of the vit encoder
  • Fine-tuning without loading position embedding from pre-trained model
  • Vit MAE reconstruction size mismatch
  • where is the tks
  • Training in cifar100
  • Hot to get the features and positional embeddig information
  • mae output normal size images
  • Any chance to have conda installation?
  • Some questions about decoder position embedding for masked tokens
  • MAE image recon
  • How to save and restore the parameters?

See more issues on GitHub

Related Packages & Articles

kornia 0.6.9

Open Source Differentiable Computer Vision Library for PyTorch

boxdetect 1.0.1

boxdetect is a Python package based on OpenCV which allows you to easily detect rectangular shapes like characters boxes on scanned forms.

pytorch-lightning 1.8.6

PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.


MediaPipe is the simplest way for researchers and developers to build world-class ML solutions and applications for mobile, edge, cloud and the web.

gluonts 0.11.6

GluonTS is a Python toolkit for probabilistic time series modeling, built around MXNet.