Vision-Transformer Directions to train and evaluate the model: Use the following command- python train.py --train_data path/to/train_data --test_data path/to/test_data --num_classes n