File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1313# Import codefiles
1414from run_experiment import run_experiment , EVALUATIONS_PER_RUN
1515from analyze_experiment import main
16- import argparse
1716
1817def analyze_experiment (experiment_name ) -> None :
1918 main (experiment = experiment_name , diagnose_pso = False )
@@ -83,7 +82,7 @@ def allocate_job(memory_per_job_gb):
8382 run_jobs .append ((model , optimizer ))
8483
8584 with multiprocessing .Pool (processes = allocate_job (1 )) as pool :
86- results = pool .map ( run_experiment , run_jobs )
85+ results = pool .starmap ( ga_comp_run_experiment , run_jobs )
8786
8887 if not all (results ):
8988 print ("Some experiments failed." )
@@ -100,7 +99,7 @@ def allocate_job(memory_per_job_gb):
10099 analyze_jobs .append ((model , optimizer ))
101100
102101 with multiprocessing .Pool (processes = allocate_job (1 )) as pool :
103- analyze_results = pool .map ( analyze_experiment , analyze_jobs )
102+ analyze_results = pool .starmap ( ga_comp_analyze_experiment , analyze_jobs )
104103
105104 if not all (analyze_results ):
106105 print ("Some analyses failed." )
You can’t perform that action at this time.
0 commit comments