From ae1c9180165d9264e9cfe152bb64164926b5ddd2 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Sun, 3 Mar 2024 12:18:25 +0100
Subject: [PATCH] Update.
---
tiny_vae.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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
--
2.39.5