Jan Hajek Jan Hajek glenn-jocher commited on
Commit
efa4946
1 Parent(s): a82dce7

W&B entity support (#2298)

Browse files

* W&B entity support

* shorten wandb_entity to entity

Co-authored-by: Jan Hajek <jan.hajek@gmail.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Files changed (1) hide show
  1. train.py +2 -0
train.py CHANGED
@@ -134,6 +134,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
134
  wandb_run = wandb.init(config=opt, resume="allow",
135
  project='YOLOv5' if opt.project == 'runs/train' else Path(opt.project).stem,
136
  name=save_dir.stem,
 
137
  id=ckpt.get('wandb_id') if 'ckpt' in locals() else None)
138
  loggers = {'wandb': wandb} # loggers dict
139
 
@@ -467,6 +468,7 @@ if __name__ == '__main__':
467
  parser.add_argument('--log-artifacts', action='store_true', help='log artifacts, i.e. final trained model')
468
  parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')
469
  parser.add_argument('--project', default='runs/train', help='save to project/name')
 
470
  parser.add_argument('--name', default='exp', help='save to project/name')
471
  parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
472
  parser.add_argument('--quad', action='store_true', help='quad dataloader')
 
134
  wandb_run = wandb.init(config=opt, resume="allow",
135
  project='YOLOv5' if opt.project == 'runs/train' else Path(opt.project).stem,
136
  name=save_dir.stem,
137
+ entity=opt.entity,
138
  id=ckpt.get('wandb_id') if 'ckpt' in locals() else None)
139
  loggers = {'wandb': wandb} # loggers dict
140
 
 
468
  parser.add_argument('--log-artifacts', action='store_true', help='log artifacts, i.e. final trained model')
469
  parser.add_argument('--workers', type=int, default=8, help='maximum number of dataloader workers')
470
  parser.add_argument('--project', default='runs/train', help='save to project/name')
471
+ parser.add_argument('--entity', default=None, help='W&B entity')
472
  parser.add_argument('--name', default='exp', help='save to project/name')
473
  parser.add_argument('--exist-ok', action='store_true', help='existing project/name ok, do not increment')
474
  parser.add_argument('--quad', action='store_true', help='quad dataloader')