/Pytorch

Primary LanguageJupyter Notebook

Pytorch

Classifying images of everyday objects using a neural network

The ability to try many different neural network architectures to address a problem is what makes deep learning really powerful, especially compared to shallow learning techniques like linear regression, logistic regression etc.


image

Explore the CIFAR10 dataset: https://www.cs.toronto.edu/~kriz/cifar.html

The CIFAR-10 dataset

The CIFAR-10 dataset consists of 60000 32x32 colour images in 10 classes, with 6000 images per class.
There are 50000 training images and 10000 test images.

The dataset is divided into five training batches and one test batch, each with 10000 images. The test batch contains exactly 1000 randomly-selected images from each class. The training batches contain the remaining images in random order, but some training batches may contain more images from one class than another. Between them, the training batches contain exactly 5000 images from each class. image

Set up a training pipeline to train a neural network on a GPU

Experiment with different network architectures & hyperparameters


Find these notebooks useful for reference, as you work through this notebook:

https://jovian.ml/aakashns/04-feedforward-nn
https://jovian.ml/aakashns/fashion-feedforward-minimal