This is a Python3 / Tensorflow implementation of a convolutional network model for MNIST dataset.

To run this code you need the following:

• Python3.3+
• Numpy
• TensorFlow

Use the main.py script to train the model. To train the default model on MNIST simply use:

python3 main.py

This code will automatically output the predictions of test-image, which is in the mnist_data/ folder.

To use the pretrained model append the -l flag after the command.

To only predict the test images using the pretrained model append the -l -e 0 flag after the command.

Train/Validation: 60000/5000

Train the model for 30 epochs.