Rbm experiment - Githubissues

google-research / discs

DISCS: The code base for the Benchmark for Discrete Sampling

Apache License 2.0

43 stars 7 forks source link

Closed KatayoonGoshvadi closed 1 year ago

KatayoonGoshvadi commented 1 year ago

Hi Hanjun, please find below the updated code of RBM.

The code has been tested to train a model on MNIST dataset, store the trained params, load the params in the sampling experiment and call sampling algs on the RBM model. Thanks and I look forward to your comments!