Update model.

vae.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4cd356f31255f04dd088f5711c8a32863ee0e4a2a58b63b02e1dac8753533bb9
+oid sha256:af795c0c5461e246c1a45230282fd7754a1572b983c0af3d08353edfb66874fc
 size 3480699

vae.py

@@ -59,6 +59,6 @@ class VAE(nn.Module):
     def forward(self, x):
         x = x.reshape((-1, 1, *self.image_dim))
         mu, sigma = self.encoder(x)
-        c = mu + sigma * torch.randn_like(sigma)
+        c = mu + sigma * torch.randn_like(sigma) * 2
         xhat = self.decoder(c)
         return xhat, mu, sigma