PyTorch-VAE icon indicating copy to clipboard operation
PyTorch-VAE copied to clipboard

Input and decoder output value ranges

Open royhj opened this issue 3 years ago • 1 comments

Thank you for sharing this repo! In VanillaVAE (and maybe others) I was wondering about the choice of using tanh as the final activation for outputs (that has a range of [-1,1]) without normalizing the input images to the same range [-1,1] (in the dataset transforms). Wouldn't it make the reconstruction loss work much harder, opposed to either normalizing the input or using a final activation like sigmoid?

royhj avatar Dec 28 '21 18:12 royhj

I had the same question. I think the output should be sigmoid and not tanh.

HareshKarnan avatar Jun 30 '22 17:06 HareshKarnan