Make wandb project name configurable (#279)

mlfoundations · Dec 8, 2022 · 00ec2e7 · 00ec2e7
1 parent 23ddc99
commit 00ec2e7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/src/training/main.py b/src/training/main.py
@@ -248,7 +248,7 @@ def main(args):
             args.val_sz = data["val"].dataloader.num_samples
         # you will have to configure this for your project!
         wandb.init(
-            project="open-clip",
+            project=args.wandb_project_name,
             name=args.name,
             notes=args.wandb_notes,
             tags=[],

diff --git a/src/training/params.py b/src/training/params.py
@@ -260,6 +260,12 @@ def parse_args(args):
         type=str,
         help="Notes if logging with wandb"
     )
+    parser.add_argument(
+        "--wandb-project-name",
+        type=str,
+        default='open-clip',
+        help="Name of the project if logging with wandb.",
+    )
     parser.add_argument(
         "--debug",
         default=False,