/Model-Performance-and-Tuning-Analysis-on-CIFAR

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, and Optuna. MLflow, DVC, YAML files and the Hydra framework are used.

Primary LanguagePythonMIT LicenseMIT

Stargazers