(args)
| 11 | from train import train |
| 12 | |
| 13 | def parse(args): |
| 14 | parser = argparse.ArgumentParser( |
| 15 | description='wandb_usage', formatter_class=argparse.RawDescriptionHelpFormatter) |
| 16 | parser.add_argument("--team_name", type=str) |
| 17 | parser.add_argument("--project_name", type=str) |
| 18 | parser.add_argument("--experiment_name", type=str) |
| 19 | parser.add_argument("--scenario_name", type=str) |
| 20 | parser.add_argument("--wandb_log_path", type=str, default="../../wandb_results/") |
| 21 | parser.add_argument("--seed",type=int,default=0) |
| 22 | parser.add_argument("--sweep_worker_num",type=int,default=2) |
| 23 | all_args = parser.parse_known_args(args)[0] |
| 24 | return all_args |
| 25 | |
| 26 | class SweepWorker(): |
| 27 | def __init__(self, args, sweep_id): |
no outgoing calls
no test coverage detected