Fix distribution names in the VAE tutorial write-up
" return mu_img\n",
"```\n",
"\n",
- "Given a latent code $z$, the forward call of `Decoder` returns the parameters for a (diagonal) gaussian distribution in image space. Since each image is of size\n",
+ "Given a latent code $z$, the forward call of `Decoder` returns the parameters for a Bernoulli distribution in image space. Since each image is of size\n",
"$28\\times28=784$, `mu_img` is of size `batch_size` x 784.\n",
"\n",
"Next we define a PyTorch module that encapsulates our encoder network:\n",
" return z_mu, z_sigma\n",
"```\n",
"\n",
- "Given an image $\\bf x$ the forward call of `Encoder` returns a mean and covariance that together parameterize a distribution in latent space.\n",
+ "Given an image $\\bf x$ the forward call of `Encoder` returns a mean and covariance that together parameterize a (diagonal) Gaussian distribution in latent space.\n",
"\n",
"With our encoder and decoder networks in hand, we can now write down the stochastic functions that represent our model and guide. First the model: \n",
"\n",
