From: François Fleuret Date: Sun, 3 Mar 2024 11:18:25 +0000 (+0100) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=ae1c9180165d9264e9cfe152bb64164926b5ddd2;p=pytorch.git Update. --- 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