Spaces:
Sleeping
Sleeping
Andrei Cozma
commited on
Commit
·
401880e
1
Parent(s):
ec64bb9
Updates
Browse files- run_tests.py +7 -2
run_tests.py
CHANGED
@@ -9,9 +9,14 @@ num_tests = 25
|
|
9 |
|
10 |
def run_test(args):
|
11 |
os.system(
|
12 |
-
f"python3 MonteCarloAgent.py --train --gamma {args[0]} --epsilon {args[1]} --wandb_project cs581 --wandb_job_type params"
|
13 |
)
|
14 |
|
15 |
|
16 |
with multiprocessing.Pool(8) as p:
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
def run_test(args):
|
11 |
os.system(
|
12 |
+
f"python3 MonteCarloAgent.py --train --gamma {args[0]} --epsilon {args[1]} --wandb_project cs581 --wandb_job_type params --wandb_run_name_suffix {args[2]}"
|
13 |
)
|
14 |
|
15 |
|
16 |
with multiprocessing.Pool(8) as p:
|
17 |
+
# make all the tests
|
18 |
+
tests = []
|
19 |
+
for gamma in vals_gamma:
|
20 |
+
for eps in vals_eps:
|
21 |
+
tests.extend((gamma, eps, i) for i in range(num_tests))
|
22 |
+
p.map(run_test, tests)
|