Tensorflow implmentation of a VAE with Bininary Concrete (BinConcrete) latent distribution, based on:
"The concrete distribution: A continuous relaxation of discrete random variables" Maddison, Chris J., Andriy Mnih, and Yee Whye Teh, ICLR, 2017
You can simply run this code by
python experiment.py
This code has been tested with
python 3.6tensorflow 2.1.0tensorflow-probability 0.9.0matplotlib 3.1.2
Install conda environment via
conda env create -f environment.yml
source activate binary_vae
Please be warned that this repository is not going to be maintained regularly.