question for class VAE(AE) #6
Description
In the vae_models.py file, the code in the class VAE(AE) is as follows:
h = self.encode(x)
mu = self.encode_mu(h)
logvar = self.encode_si(h)
z = self._reparameterize(mu, logvar)
recon = self.decode(mu)
Shouldn't the input to our decoder be the reparameterized variable z instead of mu?
"recon = recon if self.exclude_imp else recon" should be "recon = recon*m if self.exclude_imp else recon"