Model performance and tuning analysis conducted on the CIFAR10 and CIFAR100 datasets. Convolutional Neural Network (CNN), Gated Multilayer Perceptron (gMLP), and Vision Transformer (ViT) model architectures are utilized. The study is built using PyTorch, PyTorch Lightning for clean and concise code and Optuna for hyperparameter tuning.
-
Updated
Jul 16, 2024 - Python