AnalysisBySynthesis icon indicating copy to clipboard operation
AnalysisBySynthesis copied to clipboard

About training code

Open d12306 opened this issue 5 years ago • 4 comments

@lukas-schott , thank you for your implementation, can you also show the training code in order to me to evaluate it on other datasets?

Thanks

d12306 avatar Aug 27 '19 08:08 d12306

As our training code is intertwined with other projects, we do not plan on releasing it.

However, it should be straight forward to train one VAE for each class. All relevant hyperparameters should be in the paper. Otherwise, I am happy to answer any questions regarding the training procedure.

lukas-schott avatar Aug 28 '19 14:08 lukas-schott

Thanks, I will try to train it and see what happens.

d12306 avatar Aug 29 '19 03:08 d12306

@lukas-schott Thanks for your awesome project and code! I am trying to train VAEs using the code. The loss function I used is ELBOs() in the abs/abs_models/loss_functions.py. However, the trained models failed to generate realistic digits. The loss function ELBOs() return elbo / (n_ch * nx * ny), which is not correct, in my opinion. There is no reason to divide elbo by the input dimensions. I appreciate your comments.

JindongGu avatar Dec 06 '19 22:12 JindongGu

dividing the ELBO by the input dimension in https://github.com/bethgelab/AnalysisBySynthesis/blob/master/abs_models/loss_functions.py#L37 is equivalent to adjusting the learning rate and can be seen as a hyperparameter.

lukas-schott avatar Dec 09 '19 09:12 lukas-schott