From 09f05938e0870d79ff1f48efb86bfbcef19d743a Mon Sep 17 00:00:00 2001 From: kisnikser Date: Tue, 26 Nov 2024 14:01:05 +0300 Subject: [PATCH] small changes --- demo/vae_invertible_gaussian.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/vae_invertible_gaussian.py b/demo/vae_invertible_gaussian.py index b2c6fca..ddc4f6c 100644 --- 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