Skip to content
Snippets Groups Projects
Commit d6593c4a authored by Pat Alt's avatar Pat Alt
Browse files

no clue what's up

parent 4e879791
No related branches found
No related tags found
1 merge request!7669 initial run including fmnist lenet and new method
......@@ -77,12 +77,6 @@ function run_benchmark(exper::Experiment, model_dict::Dict)
continue
end
@info "Benchmarking factual=$(factual) ▶️ target=$(target)."
# Progress bars are only shown on the root process:
if !(is_multi_processed(exper) && MPI.Comm_rank(exper.parallelizer.comm) != 0)
verbose = true
else
verbose = false
end
bmk = benchmark(
counterfactual_data;
models=model_dict,
......@@ -99,7 +93,7 @@ function run_benchmark(exper::Experiment, model_dict::Dict)
push!(bmks, bmk)
end
end
bmk = reduce(vcat, bmks)
return bmk, generator_dict
final_bmk = reduce(vcat, bmks)
return final_bmk, generator_dict
end
......@@ -94,6 +94,10 @@ function run_experiment(exper::Experiment; save_output::Bool=true, only_models::
# Benchmark
benchmark!(outcome, exper)
if is_multi_processed(exper)
MPI.Barrier(exper.parallelizer.comm)
end
# Save data:
if save_output && !(is_multi_processed(exper) && MPI.Comm_rank(exper.parallelizer.comm) != 0)
Serialization.serialize(joinpath(exper.output_path, "$(exper.save_name)_outcome.jls"), outcome)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment