(args)
| 24 | |
| 25 | |
| 26 | def parse(args): |
| 27 | parser = argparse.ArgumentParser( |
| 28 | description='wandb_usage', formatter_class=argparse.RawDescriptionHelpFormatter) |
| 29 | parser.add_argument("--team_name", type=str) |
| 30 | parser.add_argument("--project_name", type=str) |
| 31 | parser.add_argument("--experiment_name", type=str) |
| 32 | parser.add_argument("--scenario_name", type=str) |
| 33 | parser.add_argument("--seed",type=int,default=0) |
| 34 | all_args = parser.parse_known_args(args)[0] |
| 35 | return all_args |
| 36 | |
| 37 | def test_pytorch(args): |
| 38 |