group.add_argument('--wandb-exp-name',type=str,default='', 683683 help='The wandb experiment name.') 684- 684+ group.add_argument('--wandb-save-dir',type=str,default='', 685+ help='Path to save the wandb results locally.') ...
When working on a distributed setup wandb seems to be using the /tmp directory to store some things and the directory specified in the dir argument. Here is how I initialize the wandb run ifglobal_rank==0:importwandbos.makedirs(args.output_dir,exist_ok=True)wandb.init(project="geo-pretrai...