| 10 | from load_data import load,read,preprocess |
| 11 | |
| 12 | def parse(args): |
| 13 | parser = argparse.ArgumentParser( |
| 14 | description='wandb_usage', formatter_class=argparse.RawDescriptionHelpFormatter) |
| 15 | parser.add_argument("--team_name", type=str) |
| 16 | parser.add_argument("--project_name", type=str) |
| 17 | parser.add_argument("--experiment_name", type=str) |
| 18 | parser.add_argument("--scenario_name", type=str) |
| 19 | parser.add_argument("--wandb_log_path", type=str, default="../../wandb_results/") |
| 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 set_wandb(all_args): |
| 25 | run_dir = Path("../../../wandb_results") / all_args.project_name / all_args.experiment_name |