(sweep_worker_num,args,sweep_id)
| 43 | |
| 44 | |
| 45 | def make_program(sweep_worker_num,args,sweep_id): |
| 46 | program = lp.Program('wandb_sweep') |
| 47 | with program.group('sweep_worker'): |
| 48 | for _ in range(sweep_worker_num): |
| 49 | program.add_node(lp.CourierNode(SweepWorker,args,sweep_id)) |
| 50 | return program |
| 51 | |
| 52 | def set_wandb(all_args): |
| 53 | run_dir = Path("../../../wandb_results") / all_args.project_name / all_args.experiment_name |