Hardcode var_scale 1.

astro-informatics · Oct 24, 2023 · 49482d3 · 49482d3
1 parent 0e79082
commit 49482d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/harmonic/model_nf.py b/harmonic/model_nf.py
@@ -27,7 +27,7 @@ def loss(params):
             log_det = model.apply(
                 {"params": params, "variables": variables},
                 batch,
-                # var_scale=1.0,
+                var_scale=1.0,
                 method=model.log_prob,
             )
             return -jnp.mean(log_det)