在MNIST数据集上训练RBM模型

这个例子使用MNIST数据集来训练一个RBM模型。RBM模型及其超参数参考Hinton的论文中的设定。

操作说明

• 下载预处理的MNIST数据集。

• 开始训练，

    python train.py mnist.pkl.gz
  

  默认情况下，训练代码将在CPU上运行。 要在GPU卡上运行它，请使用附加参数启动程序，

    python train.py mnist.pkl.gz --use_gpu