Fix LL comparison of before/after saving

This issue was originally discovered by Fabrizio Ventola (https://github.com/fabriziov). Signed-off-by: Steven Lang <[email protected]>
cambridge-mlg · Feb 22, 2021 · 2135f8b · 2135f8b
1 parent 47acddc
commit 2135f8b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/demo_mnist.py b/src/demo_mnist.py
@@ -242,9 +242,9 @@
 test_ll = EinsumNetwork.eval_loglikelihood_batched(einet, test_x, batch_size=batch_size)
 print()
 print("Log-likelihoods before saving --- train LL {}   valid LL {}   test LL {}".format(
-        train_ll / train_N,
-        valid_ll / valid_N,
-        test_ll / test_N))
+        train_ll_before / train_N,
+        valid_ll_before / valid_N,
+        test_ll_before / test_N))
 print("Log-likelihoods after saving  --- train LL {}   valid LL {}   test LL {}".format(
         train_ll / train_N,
         valid_ll / valid_N,