YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

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
    
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support