gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

bug-fix in local_batch_run_coach and rename to run_multiple_seeds

Browse Source
This commit is contained in:
Gal Leibovich
2018-08-23 14:38:42 +03:00
parent a4471389a4
commit 9bb7bd2e9c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rl_coach/local_batch_run_coach.py → rl_coach/run_multiple_seeds.py
View File

@@ -90,14 +90,14 @@ if __name__ == "__main__":
gpu_list = force_list(gpu) gpu_list = force_list(gpu)
curr_gpu_idx = 0 curr_gpu_idx = 0
for level in levels: for level in levels:
if dir_prefix != "":
dir_prefix += "_"
for seed in range(num_seeds): for seed in range(num_seeds):
# select the next gpu for this run # select the next gpu for this run
set_gpu(gpu_list[curr_gpu_idx]) set_gpu(gpu_list[curr_gpu_idx])
command = ['python3', 'rl_coach/coach.py', '-ns', '-p', '{}'.format(preset), command = ['python3', 'rl_coach/coach.py', '-ns', '-p', '{}'.format(preset),
'--seed', '{}'.format(seed), '-n', '{}'.format(num_workers)] '--seed', '{}'.format(seed), '-n', '{}'.format(num_workers)]
if dir_prefix != "":
dir_prefix += "_"
if args.use_cpu: if args.use_cpu:
command.append("-c") command.append("-c")
if args.evaluation_worker: if args.evaluation_worker: