Skip to content

VaishKS/CIFAR10_CNN_PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 

Repository files navigation

CIFAR10 Image Classification using CNN with PyTorch


About

This repository contains a custom CNN trained on CIFAR-10 dataset. This model achieves a Training accuracy of 74.46% and a test accuracy of 74.31%.

Next Steps

  1. Try chaging the hyperparameters to achieve a higher accuracy within fewer epochs.
  2. Try adding more convolutional layers, or increasing the number of channels in each convolutional layer
  3. Try using a feedforward neural network and see what's the maximum accuracy you can achieve

Releases

No releases published

Packages

No packages published