X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=tiny_vae.py;fp=tiny_vae.py;h=405c103ca8304098e5380e85508274cc9e3606df;hp=cba42e113cdf1bdf5d836f8065a76bf600cc8abe;hb=ae1c9180165d9264e9cfe152bb64164926b5ddd2;hpb=dacb655fd1541d0e08e53ba4c3f15f47baab99cc

diff --git a/tiny_vae.py b/tiny_vae.py
index cba42e1..405c103 100755
--- a/tiny_vae.py
+++ b/tiny_vae.py
@@ -141,7 +141,7 @@ class ImageGivenLatentNet(nn.Module):
     def forward(self, z):
         output = self.model(z.view(z.size(0), -1, 1, 1))
         mu, log_var = output[:, 0:1], output[:, 1:2]
-        log_var.flatten(1)[...] = log_var.flatten(1)[:, :1]
+        # log_var.flatten(1)[...] = log_var.flatten(1)[:, :1]
         return mu, log_var