Skip to main content

PyTorch: CIFAR-10 Demonstration

Publish to PyPI Deploy Documentation

A progressive deep learning tutorial for image classification on the CIFAR-10 dataset using PyTorch. This project demonstrates the evolution from basic deep neural networks to optimized convolutional neural networks with data augmentation. It also provides a set of utility functions as a PyPI package for use in other projects.

View on PyPI | Documentation

Installation

Install the helper tools package locally in editable mode to use in this repository:

pip install -e .

Or install from PyPI to use in other projects:

pip install cifar10_tools

Project overview

This repository contains a series of Jupyter notebooks that progressively build more sophisticated neural network architectures for the CIFAR-10 image classification task. Each notebook builds upon concepts from the previous one, demonstrating key deep learning techniques.

Notebooks

Notebook Description
01-DNN.ipynb Deep Neural Network - Baseline fully-connected DNN classifier using nn.Sequential. Establishes a performance baseline with a simple architecture.
02-CNN.ipynb Convolutional Neural Network - Introduction to CNNs with convolutional and pooling layers using nn.Sequential. Demonstrates the advantage of CNNs over DNNs for image tasks.
03-RGB-CNN.ipynb RGB CNN - CNN classifier that utilizes full RGB color information instead of grayscale, improving feature extraction from color images.
04-optimized-CNN.ipynb Hyperparameter Optimization - Uses Optuna for automated hyperparameter tuning to find optimal network architecture and training parameters.
05-augmented-CNN.ipynb Data Augmentation - Trains the optimized CNN architecture with image augmentation techniques for improved generalization and robustness.

Requirements

  • Python >=3.10, <3.13
  • PyTorch >=2.0
  • torchvision >=0.15
  • numpy >=1.24

License

This project is licensed under the GPLv3 License - see the LICENSE file for details.

Release files for cifar10-tools 0.5.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for cifar10-tools 0.5.5
File Size Uploaded
cifar10_tools-0.5.5.tar.gz 11.8 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for cifar10-tools 0.5.5
File Interpreter ABI Platform
cifar10_tools-0.5.5-py3-none-any.whl Python 3 none any Details

Total release size: 25.3 kB

Release files / cifar10_tools-0.5.5.tar.gz

Download URL cifar10_tools-0.5.5.tar.gz
Size 11.8 kB
Tags Source
SHA-256 checksum
How to use checksums
b97aa2275715d46066f14b067aefb4673a9d35b45ba475e5035b243ac007c2a8
BLAKE2b-256 checksum
How to use checksums
bc9e1beedab86656c6b4edae47cd3913a34e37f04e0766e580486bd2040ac237
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.7

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Feb 5, 2026.

Transparency log

Release files / cifar10_tools-0.5.5-py3-none-any.whl

Download URL cifar10_tools-0.5.5-py3-none-any.whl
Size 13.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
35d2d4c64abffb81673a6375eec2a06ac0139ada0e36077a2c280cc71d400c18
BLAKE2b-256 checksum
How to use checksums
3ea837903fc1b509a13f4f671130f9be147857c816e3620970ccf6be739ec816
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
Yes
Uploaded via twine/6.1.0 CPython/3.13.7

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

PyPI Publish Attestation

PyPI verified that this artifact, at this checksum, originated from the publisher listed below.

Signed by GitHub Actions, verified by PyPI on Feb 5, 2026.

Transparency log

Release history Release notifications | RSS feed

This release

0.5.5 This release

2 release files

0.5.4

2 release files

0.5.3

2 release files

0.5.2

2 release files

0.5.0

2 release files

0.4.0

2 release files

0.3.0

2 release files

0.2.0

2 release files

0.1.0

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page