Skip to content

Trained a convolutional neural network using PyTorch and achieved 88% test accuracy.

Notifications You must be signed in to change notification settings

davebanerjee/CIFAR-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

CIFAR-Classification

I wanted to learn how to Convolutional Neural Networks worked, so I created my own model to classify images from the CIFAR-10 dataset. I attempted to document my learning progress, so maybe others may find this useful. I thought the most useful part of this exercise was learning how to load data (without crashing my computer) and apply transforms to the data without using PyTorch's built-in DataLoader.

Ultimately, I was able to achieve 88% test accuracy, which is pretty close to the human accuracy benchmark of ~94%!

About

Trained a convolutional neural network using PyTorch and achieved 88% test accuracy.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published