huyvnphan/PyTorch_CIFAR10
Pretrained CNN models (VGG, ResNet, DenseNet, MobileNet, GoogleNet, Inception) trained on CIFAR-10 with downloadable weights.

This repository adapts TorchVision’s popular CNN architectures for the CIFAR-10 dataset by modifying input dimensions, filter sizes, strides, and padding to accommodate the 32x32 images and 10 classes. It provides both the training code using PyTorch-Lightning and the trained model weights for immediate use. Supported models include VGG, ResNet, DenseNet, MobileNetV2, GoogleNet, and InceptionV3, with validation accuracies ranging from 92% to 94%.