save elba every 2nd iteration

cbg-ethz · Nov 25, 2024 · 0d9481d · 0d9481d
1 parent 3286287
commit 0d9481d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/viloca/local_haplotype_inference/learn_error_params/cavi.py b/viloca/local_haplotype_inference/learn_error_params/cavi.py
@@ -161,7 +161,8 @@ def run_cavi(
             state_init_dict,
             state_curr_dict,
         )
-
+        if iter % 2 == 0:
+            history_elbo.append(elbo)
         if iter % (min_number_iterations-1) == 0:
             history_elbo.append(elbo)
             history_mean_log_pi.append(state_curr_dict["mean_log_pi"])

diff --git a/viloca/local_haplotype_inference/use_quality_scores/cavi.py b/viloca/local_haplotype_inference/use_quality_scores/cavi.py
@@ -142,6 +142,8 @@ def run_cavi(
             state_curr_dict,
         )
 
+        if iter % 2 == 0:
+            history_elbo.append(elbo)
         if iter % (min_number_iterations-1) == 0:
             history_elbo.append(elbo)
             history_mean_log_pi.append(state_curr_dict["mean_log_pi"])