mnist

MNIST classification

Trains a simple convolutional network on the MNIST dataset.

Requirements

• TensorFlow dataset mnist will be downloaded and prepared automatically, if necessary

Example output

I0527 09:46:48.913452 139989059204928 mnist_lib.py:128] train epoch: 10, loss: 0.0064, accuracy: 99.82
I0527 09:46:50.714237 139989059204928 mnist_lib.py:178] eval epoch: 10, loss: 0.0327, accuracy: 99.07

How to run

python mnist_main.py --model_dir=/tmp/mnist

Overriding Hyperparameter configurations

MNIST example allows specifying a hyperparameter configuration by the means of setting --config flag. Configuration flag is defined using config_flags. config_flags allows overriding configuration fields. This can be done as follows:

python mnist_main.py \
--model_dir=/tmp/mnist --config=configs/default.py \
--config.learning_rate=0.05 --config.num_epochs=5