fix tolerance update

caravagnalab · Jan 15, 2025 · d6a9ef4 · d6a9ef4
1 parent 6492b5a
commit d6a9ef4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/R/fit_variational_h.R b/R/fit_variational_h.R
@@ -51,13 +51,15 @@ fit_variational_h <- function(input_data, max_attempts = 2, initialization = NUL
 
     retries <- 0
 
+    tol_rel_obj <- tolerance
+
     while (!fit_successful && retries < 3) {  # Set a max number of retries for each attempt
       message("Retries ", retries, " of ", 3)
 
       # Increment total_attempts at the start of each retry
       total_attempts <- total_attempts + 1
 
-
+      
        result <- tryCatch({
         # Attempt variational inference
         print(list(initialization))