pythonfix
vit-pytorch 1.8.5
Vision Transformer (ViT) - Pytorch
Vision Transformer (ViT) - Pytorch
Stars: 20017, Watchers: 20017, Forks: 3000, Open Issues: 134The lucidrains/vit-pytorch repo was created 4 years ago and the last code push was Yesterday.
The project is extremely popular with a mindblowing 20017 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
- Author
- Phil Wang
- License
- MIT
- Homepage
- https://github.com/lucidrains/vit-pytorch
- PyPi:
- https://pypi.org/project/vit-pytorch/
- GitHub Repo:
- https://github.com/lucidrains/vit-pytorch
Classifiers
- Scientific/Engineering/Artificial Intelligence
Related Packages
Errors
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 134 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 train.py? 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?
Related Packages & Articles
kornia 0.7.3
Open Source Differentiable Computer Vision Library for PyTorch
gluoncv 0.10.5.post0
Gluon CV Toolkit
autogluon.core 1.1.1
Fast and Accurate ML in 3 Lines of Code
boxdetect 1.0.2
boxdetect is a Python package based on OpenCV which allows you to easily detect rectangular shapes like characters boxes on scanned forms.
torchvision 0.19.1
image and video datasets and models for torch deep learning
pytorch-lightning 2.4.0
PyTorch Lightning is the lightweight PyTorch wrapper for ML researchers. Scale your models. Write less boilerplate.
mediapipe 0.10.15
MediaPipe is the simplest way for researchers and developers to build world-class ML solutions and applications for mobile, edge, cloud and the web.
keras-self-attention 0.51.0
Attention mechanism for processing sequential data that considers the context for each timestamp
gluonts 0.15.1
Probabilistic time series modeling in Python.