Implementation of the paper https://arxiv.org/pdf/2001.04296.pdf
The dataset that was used for the training and evaluation can be accessed here http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html
main_vae.py for training Beta-Vae (use with config2.py in configs)
main.py for training GAN (use with config1.py in configs)