| # Train a RBM model against MNIST dataset |
| |
| This example is to train an RBM model using the |
| MNIST dataset. The RBM model and its hyper-parameters are set following |
| [Hinton's paper](http://www.cs.toronto.edu/~hinton/science.pdf) |
| |
| ## Running instructions |
| |
| 1. Download the pre-processed [MNIST dataset](https://github.com/mnielsen/neural-networks-and-deep-learning/raw/master/data/mnist.pkl.gz) |
| |
| 2. Start the training |
| |
| python train.py mnist.pkl.gz |
| |
| By default the training code would run on CPU. To run it on a GPU card, please start |
| the program with an additional argument |
| |
| python train.py mnist.pkl.gz --use_gpu |