Skip to content

Implementing an ANN using PyTorch (under 800,000 parameters) to achieve +92% accuracy in under 100 epochs.

Notifications You must be signed in to change notification settings

Alireza-Gharavi/CIFAR-10-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Implementing a Classification task on the CIFAR-10 dataset

In this notebook, the achievement is reaching +92% accuracy on CIFAR-10 using an ANN with <800,000 parameters in under 100 epochs. The training and testing phases are performed in two different approaches:

  • One using K-fold cross-validation
  • Another without any validation dataset and only using train and test datasets for the whole process

About

Implementing an ANN using PyTorch (under 800,000 parameters) to achieve +92% accuracy in under 100 epochs.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published