practicing RTIDS and ViT in Pytorch working on it ViT Model Layers Hidden size D MLP size Heads ViT-Base 12 768 3072 12 ViT-Large 24 1024 4096 16 ViT-Huge 32 1280 5120 16 on CIFAR-10 RTIDS on CHD on ROAD