From 49482d34073b74144667e6f8b1ccd6ceb764ad6e Mon Sep 17 00:00:00 2001
From: alicjapolanska <alicja.polanska.22@ucl.ac.uk>
Date: Tue, 24 Oct 2023 11:34:19 +0100
Subject: [PATCH] Hardcode var_scale 1.

---
 harmonic/model_nf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/harmonic/model_nf.py b/harmonic/model_nf.py
index 41e2fa6c..192121f2 100644
--- 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)