This repository implements a ResNet-8 architecture for image classification on the CIFAR-10 dataset using PyTorch.
The CIFAR-10 dataset consists of 60,000 RGB images (32x32 pixels) across 10 classes:
- Airplane
- Automobile
- Bird
- Cat
- Deer
- Dog
- Frog
- Horse
- Ship
- Truck
- Implementation of a custom ResNet-8 architecture optimized for CIFAR-10.
- Training and evaluation pipelines using PyTorch.
- Visualization of training performance and model predictions.
- Python 3.x
- PyTorch
- TorchVision
- Matplotlib
Install dependencies using:
pip install torch torchvision matplotlib