From 64821a07333adff34b0d7ddfc427063770831a33 Mon Sep 17 00:00:00 2001 From: C-Achard Date: Tue, 7 May 2024 15:58:05 +0200 Subject: [PATCH] Fix WandB run names --- napari_cellseg3d/code_models/worker_training.py | 2 +- napari_cellseg3d/dev_scripts/colab_training.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/napari_cellseg3d/code_models/worker_training.py b/napari_cellseg3d/code_models/worker_training.py index 0ffec8f9..90aeeba5 100644 --- 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 index e8ca1f91..a5020fec 100644 --- 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)