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

About

No description or website provided.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published