Skip to content

Files

Latest commit

49292f3 · Nov 22, 2017

History

History
This branch is 8058 commits behind tensorflow/models:master.

mnist

MNIST in TensorFlow

This directory builds a convolutional neural net to classify the MNIST dataset using the tf.data, tf.estimator.Estimator, and tf.layers APIs.

Setup

To begin, you'll simply need the latest version of TensorFlow installed.

First convert the MNIST data to TFRecord file format by running the following:

python convert_to_records.py

Then to train the model, run the following:

python mnist.py

The model will begin training and will automatically evaluate itself on the validation data.