search

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.
Apache License 2.0
6.62k stars 1.06k forks source link

Sign issue in gaussian kernel function of MS-SSIM-VAE #41

Open tobias-haenel opened 3 years ago

tobias-haenel commented 3 years ago

The computation of the gaussian kernel is missing a sign in the exponent:https://github.com/AntixK/PyTorch-VAE/blob/8700d245a9735640dda458db4cf40708caf2e77f/models/mssim_vae.py#L204