Learn/Refresh about Vision Transformer architecture
- pytorch
- matplotlib
- einops
# create environment
python3 -m venv venv
source venv/bin/activate
pip install -r requirements.txt
Then run cells in Jupyter notebook
Implementing Vision Transformer (ViT) in PyTorch by Francesco Zuppichini