PyTorch: CIFAR-10 Demonstration
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.
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)
| File | Size | Uploaded | |
|---|---|---|---|
| cifar10_tools-0.5.5.tar.gz | 11.8 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| 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 logRelease 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