microvault / vit

๐Ÿšถโ€โ™‚๏ธ Simple ViT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Requirements

  • Generate synthetic data
  • View attention maps
  • Use torch 2x
  • Train inside docker compose with cuda
  • Build as package
  • Deploy on hugging face
  • Automated network testing
  • Use lighting, accelerator, tqdm
  • Use wandb
  • Build the model with pytorch models
  • Parameter search (ray tune)
  • View easily
  • Use mkdocs in black
  • Use rerun

About

๐Ÿšถโ€โ™‚๏ธ Simple ViT


Languages

Language:Python 98.5%Language:Dockerfile 1.5%