small changes

intsystems · Nov 26, 2024 · 09f0593 · 09f0593
1 parent 7a8ad24
commit 09f0593
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo/vae_invertible_gaussian.py b/demo/vae_invertible_gaussian.py
@@ -59,7 +59,7 @@ def parse_arguments() -> argparse.Namespace:
 N = 20  # Number of categorical distributions
 
 ### CHANGE IT TO 1/K CATEGORICAL APPROXIMATION
-loc_prior = torch.zeros(N, K - 1, device=device)
+loc_prior = torch.ones(N, K - 1, device=device)
 scale_prior = torch.ones(N, K - 1, device=device)
 
 temp = INITIAL_TEMP