tree: ec3ded474b5346bbf805b55a699a46e254d3eaf4 [path history] [tgz]

Train a RBM model using SINGA Tensor APIs

This example is to train a RBM model over the MNIST dataset using SINGA Python Tensor APIs. The RBM model and its hyper-parameters are set following Hinton's paper

Running instructions

  1. Download the pre-processed MNIST dataset

  2. Start the training

     python 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 mnist.pkl.gz --use_gpu