Pytorch implementation of Variational Autoencoder.

https://github.com/Lyusungwon/generative_models_pytorch

Reference

  • https://github.com/pytorch/examples/tree/master/vae

Note

  • Divided encoder and decoder

Results

Config

  • model: 180824083213_vae_5000_200_1e-05_28_28_400_16_1
  • epochs 5000 batch-size 200 lr 1e-5 hidden-size 400 latent-size 16 L 1

Train

image

  • KLD 24.13 Recon 76.11 Train loss 100.2

Test

image

  • KLD 24.08 Recon 69.89 Test loss 93.87

Reconstruction

image

Sampling

image