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
b81df9a
..
cba42e1
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]
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
return mu, log_var
@@
-234,14
+234,16
@@
z = sample_gaussian(param_q_Z_given_x)
param_p_X_given_z = model_p_X_given_z(z)
x = sample_gaussian(param_p_X_given_z)
save_image(x, "output.png")
param_p_X_given_z = model_p_X_given_z(z)
x = sample_gaussian(param_p_X_given_z)
save_image(x, "output.png")
+save_image(param_p_X_given_z[0], "output_mean.png")
# Generate a bunch of images
z = sample_gaussian(
# Generate a bunch of images
z = sample_gaussian(
-
param_p_Z[0].expand(x.size(0), -1), param_p_Z[1].expand(x.size(0), -1
)
+
(param_p_Z[0].expand(x.size(0), -1), param_p_Z[1].expand(x.size(0), -1)
)
)
param_p_X_given_z = model_p_X_given_z(z)
x = sample_gaussian(param_p_X_given_z)
save_image(x, "synth.png")
)
param_p_X_given_z = model_p_X_given_z(z)
x = sample_gaussian(param_p_X_given_z)
save_image(x, "synth.png")
+save_image(param_p_X_given_z[0], "synth_mean.png")
######################################################################
######################################################################