uf

experiments/experiment.jl

@@ -116,6 +116,11 @@ function run_experiment(exper::Experiment; save_output::Bool=true, only_models::
         meta(outcome; save_output=true)
     end
 
+    # Final barrier:
+    if is_multi_processed(exper)
+        MPI.Barrier(exper.parallelizer.comm)
+    end
+
     return outcome
 
 end

experiments/grid_search.jl

@@ -27,7 +27,9 @@ function grid_search(
     outcomes = Dict{Any,Any}()
     
     # Search:
+    counter = 1
     for tuning_params in grid
+        @info "Running experiment $(counter)/$(length(grid)) with tuning parameters: $(tuning_params)"
         outcome = run_experiment(
             counterfactual_data, test_data;
             save_output=false,
@@ -38,6 +40,7 @@ function grid_search(
             kwargs...,
         )
         outcomes[tuning_params] = outcome
+        counter += 1
     end
 
     # Save: