improved wandb logging logic

clemsgrs · Oct 24, 2024 · f5acaad · f5acaad
1 parent 180659f
commit f5acaad
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dinov2/utils/utils.py b/dinov2/utils/utils.py
@@ -128,8 +128,9 @@ def initialize_wandb(
     cfg: DictConfig,
     key: Optional[str] = "",
 ):
-    command = f"wandb login {key}"
-    subprocess.call(command, shell=True)
+    if not key:
+        key = input("Enter your W&B API key: ")
+    wandb.login(key=key)
     if cfg.wandb.tags is None:
         tags = []
     else: