Skip to content

Pauljanson002/VisionTransformers

Repository files navigation

Vision Transformers

Train a model

python train.py --epochs 100 --model vit --savename 123

Use the above code to train a model
Models available

  • Vit
  • Naive vit

Validate a model

python validate.py --model vit --path vit.pt

This code is used to validate a model to find its test and validation accuracy

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages