mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #2398 from huggingface/caojiaolong-main

Browse Source 
Merging wandb project name chages w/ addition
This commit is contained in:
Ross Wightman 2025-01-08 10:15:09 -08:00 committed by GitHub
commit c4fb98f399
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
train.py
View File

@ -390,6 +390,8 @@ group.add_argument('--use-multi-epochs-loader', action='store_true', default=Fal
help='use the multi-epochs-loader to save time at the beginning of every epoch') help='use the multi-epochs-loader to save time at the beginning of every epoch')
group.add_argument('--log-wandb', action='store_true', default=False, group.add_argument('--log-wandb', action='store_true', default=False,
help='log training and validation metrics to wandb') help='log training and validation metrics to wandb')
group.add_argument('--wandb-project', default=None, type=str,
help='wandb project name')
group.add_argument('--wandb-tags', default=[], type=str, nargs='+', group.add_argument('--wandb-tags', default=[], type=str, nargs='+',
help='wandb tags') help='wandb tags')
group.add_argument('--wandb-resume-id', default='', type=str, metavar='ID', group.add_argument('--wandb-resume-id', default='', type=str, metavar='ID',
@ -832,14 +834,15 @@ def main():
with open(os.path.join(output_dir, 'args.yaml'), 'w') as f: with open(os.path.join(output_dir, 'args.yaml'), 'w') as f:
f.write(args_text) f.write(args_text)
if utils.is_primary(args) and args.log_wandb: if args.log_wandb:
if has_wandb: if has_wandb:
assert not args.wandb_resume_id or args.resume assert not args.wandb_resume_id or args.resume
wandb.init( wandb.init(
project=args.experiment, project=args.wandb_project,
name=exp_name,
config=args, config=args,
tags=args.wandb_tags, tags=args.wandb_tags,
resume='must' if args.wandb_resume_id else None, resume="must" if args.wandb_resume_id else None,
id=args.wandb_resume_id if args.wandb_resume_id else None, id=args.wandb_resume_id if args.wandb_resume_id else None,
) )
else: else: