About the KLD loss term
YMBetta opened this issue · comments
Imitation commented
In the original paper, loss=recon_loss - KLD. But in your code, loss=recon_loss + KLD. I aslo found your code work in both cases. Can you please tell the idea about the loss in your code?
Imitation commented
Your implementation is consistent with origin paper, I‘m’ misunderstanding it!