From 5c2e5e36fd93ab7d31e4a20759916df6ebd606a5 Mon Sep 17 00:00:00 2001
From: Pat Alt <55311242+pat-alt@users.noreply.github.com>
Date: Sun, 17 Sep 2023 11:39:24 +0200
Subject: [PATCH] uf
---
experiments/experiment.jl | 5 +++++
experiments/grid_search.jl | 3 +++
2 files changed, 8 insertions(+)
diff --git a/experiments/experiment.jl b/experiments/experiment.jl
index b013d156..f3388c04 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 52e2c6ff..090acada 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:
--
GitLab