(sweep_worker_num,args,sweep_id)
| 35 | |
| 36 | |
| 37 | def make_program(sweep_worker_num,args,sweep_id): |
| 38 | program = lp.Program('wandb_sweep') |
| 39 | with program.group('sweep_worker'): |
| 40 | for _ in range(sweep_worker_num): |
| 41 | program.add_node(lp.CourierNode(SweepWorker,args,sweep_id)) |
| 42 | return program |
| 43 | |
| 44 | def set_wandb(all_args): |
| 45 | run_dir = Path("../../../wandb_results") / all_args.project_name / all_args.experiment_name |