Simple CNN based model to train with cifar10 dataset. The program takes several parameters: epochs, train batch size, test batch size, learning rate .. . --help call on the program to get more information.
Once the model is trained and saved (specifying the --save-model flag) the prediction code takes the path of the image to evaluate (--file flag) and gives the answer.