---

id: model-zoo-rbm-mnist title: 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

Running instructions

Download

Download the pre-processed MNIST dataset

Start the training

Please cd to singa/examples/mnist/ for the following commands

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