Fix WandB run names

C-Achard · C-Achard · commit 64821a07333a · 2024-05-07T15:58:05.000+02:00
diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py
@@ -430,7 +430,7 @@ def train(
                     wandb.finish()
                 wandb.init(
                     config=config_dict,
-                    project="CellSeg3D - WNet",
+                    project="CellSeg3D - WNet3D",
                     name=f"WNet3D_training - {utils.get_date_time()}",
                     mode=self.wandb_config.mode,
                     tags=["WNet3D", "training"],
diff --git a/napari_cellseg3d/dev_scripts/colab_training.py b/napari_cellseg3d/dev_scripts/colab_training.py
@@ -598,7 +598,7 @@ def train(
                 model_artifact = wandb.Artifact(
                     "WNet3D",
                     type="model",
-                    description="CellSeg3D WNet",
+                    description="CellSeg3D WNet3D",
                     metadata=self.config.__dict__,
                 )
                 model_artifact.add_file(save_weights_path)

Original file line number	Diff line number	Diff line change
`@@ -598,7 +598,7 @@ def train(`
`598`	`598`	`model_artifact = wandb.Artifact(`
`599`	`599`	`"WNet3D",`
`600`	`600`	`type="model",`
`601`		`- description="CellSeg3D WNet",`
	`601`	`+ description="CellSeg3D WNet3D",`
`602`	`602`	`metadata=self.config.__dict__,`
`603`	`603`	`)`
`604`	`604`	`model_artifact.add_file(save_weights_path)`