also use vae latent for second level

2023-01-30 19:51:14 +00:00
parent 2b8abd776b
commit 6e24c1f41b
5 changed files with 21 additions and 9 deletions
@@ -92,7 +92,7 @@ class PyramidLevel(tf.keras.Model):
        
        # x = x * uv_mask_high[:,:,:,tf.newaxis]

-        return x, mean, logvar
+        return x, z, mean, logvar