/mup-vit

Everything you need to reproduce "Better plain ViT baselines for ImageNet-1k" in PyTorch, and more

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers