Selected Research Paper : An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale
num_epochs = 10
batch_size = 16
exp_name = "vit_"+str(num_epochs)+"_"+str(batch_size)
!python train.py --exp-name exp_name --epochs num_epochs --batch-size batch_size
Aditya Varshney Gega Darakhvelidze