Fix WandB run names

AdaptiveMotorControlLab · May 7, 2024 · 64821a0 · 64821a0
1 parent f1b0c59
commit 64821a0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
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)