DBN-and-RBM-in-pytorch

This repository has a pytorch implementation(both CPU and GPU) for Deep Belief Networks and Restricted Boltzmann Machine

Special thanks to the following github repositorie：

https://github.com/mehulrastogi/Deep-Belief-Network-pytorch

My all work here is to solve the bug that the demo with GPU doesn't work.