Hi,
I am trying to generate samples through VAE, but after training, I can see a blob at the center of the image. I tried mse loss, bce loss but still, the problem persists.
The loss is shooting up and becomes nan values after some epoch.
I tried with mse loss
recons_loss =F.mse_loss(recons, input,reduction='mean')
kld_loss = torch.mean(-0.5 * torch.sum(1 + log_var - mu ** 2 - log_var.exp(), dim = 1), dim = 0)