From 5445b64ce1edbd955cace3bf3d275d7d8c4209f0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jes=C3=BAs=20Pineda?= Date: Sat, 18 Feb 2023 11:40:58 +0100 Subject: [PATCH] Update vae.py --- deeptrack/models/vaes/vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeptrack/models/vaes/vae.py b/deeptrack/models/vaes/vae.py index e1f10dbff..25c6e72ae 100644 --- a/deeptrack/models/vaes/vae.py +++ b/deeptrack/models/vaes/vae.py @@ -30,7 +30,7 @@ def train_step(self, data): # Sample a random point in the latent space epsilon = tf.random.normal(shape=tf.shape(z_mean)) - z = z_mean + tf.exp(z_log_var) * epsilon + z = z_mean + tf.exp(0.5 * z_log_var) * epsilon # Reconstruct the input image rdata = self.decoder(z)