Small WandB improvement

AdaptiveMotorControlLab · Feb 26, 2024 · b8e36af · b8e36af
1 parent 6df92a5
commit b8e36af
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -426,7 +426,9 @@ def train(
                 wandb.init(
                     config=config_dict,
                     project="CellSeg3D",
+                    name=f"{self.config.model_info.name}_training - {utils.get_date_time()}",
                     mode=self.wandb_config.mode,
+                    tags=[f"{self.config.model_info.name}", "training"],
                 )
 
             set_determinism(seed=self.config.deterministic_config.seed)
@@ -1117,6 +1119,8 @@ def train(
                     wandb.init(
                         config=config_dict,
                         project="CellSeg3D",
+                        name=f"{model_config.name}_supervised_training - {utils.get_date_time()}",
+                        tags=[f"{model_config.name}", "supervised"],
                         mode=self.wandb_config.mode,
                     )
                 except AttributeError:

diff --git a/napari_cellseg3d/dev_scripts/colab_training.py b/napari_cellseg3d/dev_scripts/colab_training.py
@@ -330,7 +330,9 @@ def train(
                 wandb.init(
                     config=config_dict,
                     project="CellSeg3D (Colab)",
+                    name=f"{self.config.model_info.name} training - {utils.get_date_time()}",
                     mode=self.wandb_config.mode,
+                    tags=["WNet", "Colab"],
                 )
 
             set_determinism(seed=self.config.deterministic_config.seed)