name wandb runs if a name was specified

tlpss · Nov 6, 2023 · cec24ca · cec24ca
1 parent 692698e
commit cec24ca
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keypoint_detection/tasks/train.py b/keypoint_detection/tasks/train.py
@@ -89,6 +89,7 @@ def train(hparams: dict) -> Tuple[KeypointDetector, pl.Trainer]:
 
     data_module = KeypointsDataModule(**hparams)
     wandb_logger = WandbLogger(
+        name=hparams["wandb_name"],
         project=hparams["wandb_project"],
         entity=hparams["wandb_entity"],
         save_dir=get_wandb_log_dir_path(),