Hello,
I have implemented a VAE in Pytorch with MNIST database.
I noticed that when I use the F.mse_loss, I don’t get a good reconstruction of the input images.
Here is an exmaple of the reconstruction of some input images:
mnistlr0001reg01.jpg1292×535 155 KB

But when I use the F.binary_cross_entropy loss, I get a good reconstruction:
Can you please help me understand why I can’t get a good reconstruction of the input images when using the mse loss?
I appreciate your help!