From b8e36af8fc3e3c5a37bb9fe5f4175803a72adcca Mon Sep 17 00:00:00 2001 From: C-Achard Date: Mon, 26 Feb 2024 11:12:08 +0100 Subject: [PATCH] Small WandB improvement --- napari_cellseg3d/code_models/worker_training.py | 4 ++++ napari_cellseg3d/dev_scripts/colab_training.py | 2 ++ 2 files changed, 6 insertions(+) diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py index c363fd31..41816eb1 100644 --- 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 index b2f8d114..ef5245e8 100644 --- 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)