(beta-)VAE
Tensorflow implementation of VAE and beta-VAE
Celeba
ConvNet (z_dim: 100, beta: 0.05)
Generation | Reconstruction |
---|---|
Mnist
ConvNet (z_dim: 10, beta: 0.1)
Generation | Reconstruction |
---|---|
MLP (z_dim: 10, beta: 0.1)
Generation | Reconstruction |
---|---|
Prerequisites
Examples of training
CUDA_VISIBLE_DEVICES=0 python train.py --z_dim 10 --beta 0.1 --dataset mnist --model mlp_mnist --experiment_name z10_beta0.1_mnist_mlp
CUDA_VISIBLE_DEVICES=0 python train.py --z_dim 10 --beta 0.1 --dataset mnist --model conv_mnist --experiment_name z10_beta0.1_mnist_conv
CUDA_VISIBLE_DEVICES=0 python train.py --z_dim 32 --beta 0.1 --dataset celeba --model conv_64 --experiment_name z32_beta0.1_celeba_conv