AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.
Apache License 2.0
6.46k stars 1.05k forks source link

The generated samples are not great for WAE_MMD #45

Open WonderSeven opened 2 years ago

WonderSeven commented 2 years ago

Hi, Friends, what about the quantity of your generated samples, my samples are not good-looking, I show some of them below.

Generated samples:

image

Reconstructed samples: image

AntixK commented 2 years ago

Try increading the MMD loss weight. Also, play around with different kernels - RBF and IMQ

WonderSeven commented 2 years ago

Thank you very much for your reply, I will try your suggested tips.