End-to-end Training of Deep Boltzmann Machines by Unbiased Contrastive Divergence with Local Mode Initialization
Minimum training code of unbiased Deep Boltzmann Machines for MNIST and Fashion MNIST
- torch
- tensorboard
python train.py --dataset MNISTpython train.py --dataset FashionMNISTpython train.py --dataset MNIST --bits 8python train.py --dataset FashionMNIST --bits 8