Merge pull request #170 from softmatterlab/jp/fix-vae

Update vae.py

Author: JesusPinedaC

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)