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)