| 11 | import wandb |
| 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("--seed",type=int,default=0) |
| 21 | all_args = parser.parse_known_args(args)[0] |
| 22 | return all_args |
| 23 | |
| 24 | def running_func(args,start_step): |
| 25 | |