A simple implementation of a Convolutional Neural Network (CNN) for image classification using the CIFAR-10 dataset. This repository demonstrates training, evaluating, and performing inference on image data.
- Implements a CNN model using PyTorch.
- Trains the model on the CIFAR-10 dataset.
- Supports model evaluation and prediction on test images.
- Highly modular for easy customization.
Follow these steps to set up the project:
- Clone the Repository
git clone https://github.com/sahankrt20/deeplearning2088.git cd deeplearningalgorithm