/vision-transformer

Vision Transformer using pytorch, pytorch-lightning and wandb

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers