Code for training a model on performing image classification on mnist dataset.
The first two lines of code in mnist-softmax.py get the data for you. To run the documented model pass the argument --data_dir .
Code for training a model on performing image classification on mnist dataset.
Jupyter Notebook