vae-celebA
vae-celebA copied to clipboard
Reproduce the plain VAE result
HI,
Thank you for sharing your code.
However, when I run the plain VAE code using the default parameters, I cannot reproduce the result you post on the REAMED page.
The reconstruction image and random image I generated using your code is:
Can you provide some suggestions on how to reproduce your result?
Just try to tweak the weight of 'style loss' and 'content loss' in the cost function. It's quite common for vae, though I still think the default params should avoid this collapse.
@yzwxx Thanks for the awesome repo. Can you please suggest as to how you chose the relative weight of KL loss with respect to reconstruction loss. I mean, can you kindly comment about the relative scales of losses that I should try to keep.