/cifar10-cnn-pytorch

CIFAR-10 Image Classification using CNN in PyTorch

Primary LanguageJupyter NotebookMIT LicenseMIT

CIFAR-10 Image Classification using CNN in PyTorch

  • The CNN used is a simplified version of a ResNet implemented based on the Myrtle.ai blog
  • The model achieves a test accuracy of 91.68% after 40 epochs of training

Hyperparameters tuning

  • The project with hyperparameters tuning can be found in the branch hyperparameters-tuning