This project is about using Convolutional Neural Networks for classification of images, for CINIC-10 data. The main goal of the project was to test various network structures and hyper-parameters values/methods.
- Create a directory called data and put the dataset there (with train and test folders).
main.py
- Here you can modify the code to choose which model and other parameters you want to use for training.- models- In this directory, the models are implemented.
plots.ipynb
- With this notebook, the plots and graphs for the report were created