CNN / README.md
Krippa's picture
Upload folder using huggingface_hub
ecf179e verified
|
Raw
History Blame Contribute Delete
460 Bytes

CNN CIFAR-10 Classifier (PyTorch)

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