diff --git a/experiments/experiment.jl b/experiments/experiment.jl index b013d15679728ef9f0fbe030c8e3d0155b0198dc..f3388c04723f4b99bb5dc2db51e976250ebd5292 100644 --- a/experiments/experiment.jl +++ b/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 diff --git a/experiments/grid_search.jl b/experiments/grid_search.jl index 52e2c6ff2818d5673523767689e706bc837bb3f5..090acadaf077a8677967be9b552b1b64d1f25b7c 100644 --- a/experiments/grid_search.jl +++ b/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: