- This project is the implementation of the following paper.
- Softmax as well as SVM classifier at the end of the Convolutional Neural Network is implemented and compared.
- Download the Cifar 10 batches and save it in the Cifar_Batches folder from the following link.
- It is advised to run it on a system with 8GB or above RAM.
Run
python3 model.py [-m] --model Softmax [-e] --epochs n_epochs [-b] --batch_size batch_size
Run
python3 model.py [-m] --model SVM [-e] --epochs n_epochs [-b] --batch_size batch_size
- The following results have been obtained by running it on a CPU system with 4GB RAM.