Skip to content

Commit

Permalink
Added more info to finished runs
Browse files Browse the repository at this point in the history
  • Loading branch information
ancestor-mithril committed Nov 18, 2024
1 parent b9003c4 commit 9f97b0c
Showing 1 changed file with 14 additions and 6 deletions.
20 changes: 14 additions & 6 deletions experiment_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def generate_runs():
64
]
lrs = [
0.1, 0.75, 0.5, 0.025, 0.01
0.1, 0.075, 0.05, 0.025, 0.01
]
reductions = [
"mean"
Expand Down Expand Up @@ -221,8 +221,16 @@ def generate_runs():
if last_index == -1 or last_index > len(runs):
last_index = len(runs)

with ProcessPoolExecutor(max_workers=gpu_count * processes_per_gpu) as executor:
executor.map(
run_command,
[(runs[index], envs[index], index) for index in range(run_index, last_index)],
)
with open("./logs/finished_runs.txt", "a+") as fp:
fp.write("New experiment: ")
fp.write("\n")

try:
with ProcessPoolExecutor(max_workers=gpu_count * processes_per_gpu) as executor:
executor.map(
run_command,
[(runs[index], envs[index], index) for index in range(run_index, last_index)],
)
finally:
with open("./logs/finished_runs.txt", "a+") as fp:
fp.write("\n")

0 comments on commit 9f97b0c

Please sign in to comment.