Find file History
Latest commit 247c2ed Feb 23, 2017 @mitmul mitmul committed on GitHub Update train_mnist.py

README.md

Multi-Layer Perceptron for MNIST Classification

This is a minimal example to write a feed-forward net. The code consists of three parts: dataset preparation, network and optimizer definition and learning loop. This is a common routine to write a learning process of networks with dataset that is small enough to fit into memory.

If you want to run this example on the N-th GPU, pass --gpu=N to the script.