GitHub - GnanaPrakashSG2004/MNIST-Classification: Implementation of ViT (Vision Transformer) and LeNet from scratch using PyTorch for the MNIST classification task. Ran multiple experiments to understand the impact of dataset size, network depth, optimizer, learning rate, and batch size on model performance

GnanaPrakashSG2004 / MNIST-Classification Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Implementation of ViT (Vision Transformer) and LeNet from scratch using PyTorch for the MNIST classification task. Ran multiple experiments to understand the impact of dataset size, network depth, optimizer, learning rate, and batch size on model performance