projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
| inline |
side by side
Update.
[pytorch.git]
/
tiny_vae.py
diff --git
a/tiny_vae.py
b/tiny_vae.py
index
cba42e1
..
405c103
100755
(executable)
--- 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