vae-celebA icon indicating copy to clipboard operation
vae-celebA copied to clipboard

Reproduce the plain VAE result

Open yangw1234 opened this issue 7 years ago • 2 comments

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:

train_29_3014

train_29_3014_random

Can you provide some suggestions on how to reproduce your result?

yangw1234 avatar Jan 24 '18 05:01 yangw1234

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 avatar Feb 07 '18 17:02 yzwxx

@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.

avisekiit avatar Jun 04 '19 18:06 avisekiit