Small WandB improvement

Author: C-Achard

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:

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)