README.md

metadata

title: CNN Cifar10 Classifier
emoji: 🚀
colorFrom: blue
colorTo: indigo
sdk: docker
pinned: false
license: mit

CNN Image Classification

This project implements a Convolutional Neural Network (CNN) to classify images from the CIFAR-10 dataset using PyTorch.

Structure

• src/: Source code for the model and data processing.
• main.py: Entry point for training and evaluation.

Usage

1. Install dependencies:

   pip install -r requirements.txt
   

2. Run the training pipeline:

   python main.py