support the team name in wandb projct

unicore/logging/progress_bar.py

@@ -313,8 +313,13 @@ class TensorboardProgressBarWrapper(BaseProgressBar):
         global _wandb_inited
         if not _wandb_inited and wandb_project and wandb_available:
             wandb_name = args.wandb_name or wandb.util.generate_id()
+            if "/" in wandb_project:
+                entity, project = wandb_project.split("/")
+            else:
+                entity, project = None, wandb_project
             wandb.init(
-                project=wandb_project,
+                project=project,
+                entity=entity,
                 name=wandb_name,
                 config=vars(args),
                 id=wandb_name,

unicore/options.py

@@ -175,7 +175,7 @@ def get_parser(desc, default_task="test"):
                         help='path to save logs for tensorboard, should match --logdir '
                              'of running tensorboard (default: no tensorboard logging)')
     parser.add_argument('--wandb-project', metavar='DIR', default='',
-                        help='name of wandb project, empty for no wandb logging, for wandb login, use env WANDB_API_KEY')
+                        help='name of wandb project, empty for no wandb logging, for wandb login, use env WANDB_API_KEY. You can also use team_name/project_name for project name.')
     parser.add_argument('--wandb-name', metavar='DIR', default='',
                         help='wandb run/id name, empty for no wandb logging, for wandb login, use env WANDB_API_KEY')
     parser.add_argument('--seed', default=1, type=int, metavar='N',