Initialize wandb writer.
(args)
| 33 | log_dir=os.path.join(base, SUMMARY_WRITER_DIR_NAME, name), purge_step=iteration) |
| 34 | |
| 35 | def init_wandb_writer(args): |
| 36 | """Initialize wandb writer.""" |
| 37 | import wandb |
| 38 | wandb.init( |
| 39 | project=args.wandb_project_name, |
| 40 | name=args.experiment_name, |
| 41 | dir=args.summary_dir, |
| 42 | ) |
| 43 | |
| 44 | # def print_rank0(message): |
| 45 | # if torch.distributed.is_initialized(): |