Add wandb project name argument and allow change wandb run name

Author: caojiaolong

train.py

@@ -388,6 +388,8 @@
                    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,
                    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='+',
                    help='wandb tags')
 group.add_argument('--wandb-resume-id', default='', type=str, metavar='ID',
@@ -823,9 +825,14 @@ def main():
     if utils.is_primary(args) and args.log_wandb:
         if has_wandb:
             assert not args.wandb_resume_id or args.resume
-            wandb.init(project=args.experiment, config=args, tags=args.wandb_tags,
-                       resume='must' if args.wandb_resume_id else None,
-                       id=args.wandb_resume_id if args.wandb_resume_id else None)
+            wandb.init(
+                project=args.wandb_project,
+                name=args.experiment,
+                config=args,
+                tags=args.wandb_tags,
+                resume="must" if args.wandb_resume_id else None,
+                id=args.wandb_resume_id if args.wandb_resume_id else None,
+            )
         else:
             _logger.warning(
                 "You've requested to log metrics to wandb but package not found. "