vit-pytorch 0.26.3


Vision Transformer (ViT) - Pytorch

Vision Transformer (ViT) - Pytorch

Stars: 8219, Watchers: 8219, Forks: 1298, Open Issues: 80

The lucidrains/vit-pytorch repo was created 1 years ago and was last updated 20 hours ago.
The project is extremely popular with a mindblowing 8219 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.

No  vit-pytorch  errors just yet.

Code Examples

Here are some vit-pytorch code examples and snippets.

No  vit-pytorch  code examples just yet.

GitHub Issues

The vit-pytorch package as 80 open issues on GitHub

  • Fine-tuning without loading position embedding from pre-trained model
  • Vit MAE reconstruction size mismatch
  • where is the tks
  • About different patch embedding in ViT
  • Training in cifar100
  • Hot to get the features and positional embeddig information
  • mae output normal size images
  • Any chance to have conda installation?
  • Vision Transformers for Federated Learning use-cases
  • MAE decoder pos_emb
  • Vit3D for Brain CTscans
  • Some questions about decoder position embedding for masked tokens
  • MAE image recon
  • How to save and restore the parameters?
  • ask help: what's the meaning of self.to_latent = nn.Identity() in

See more issues on GitHub

See Also

kornia 0.6.2

Open Source Differentiable Computer Vision Library for PyTorch

boxdetect 1.0.0

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