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