Remove redundant code + Colab notebook tweaks

Author: C-Achard

napari_cellseg3d/dev_scripts/colab_training.py

@@ -30,17 +30,6 @@
 VERBOSE_SCHEDULER = True
 logger.debug(f"PRETRAINED WEIGHT DIR LOCATION : {PRETRAINED_WEIGHTS_DIR}")
 
-try:
-    import wandb
-
-    WANDB_INSTALLED = True
-except ImportError:
-    logger.warning(
-        "wandb not installed, wandb config will not be taken into account",
-        stacklevel=1,
-    )
-    WANDB_INSTALLED = False
-
 
 class LogFixture:
     """Fixture for napari-less logging, replaces napari_cellseg3d.interface.Log in model_workers.
@@ -161,8 +150,7 @@ def get_colab_worker(
         wandb_config (config.WandBConfig): config for wandb
     """
     log = LogFixture()
-    worker = WNetTrainingWorkerColab(worker_config)
-    worker.wandb_config = wandb_config
+    worker = WNetTrainingWorkerColab(worker_config, wandb_config)
 
     worker.log_signal.connect(log.print_and_log)
     worker.warn_signal.connect(log.warn)

notebooks/Colab_WNet3D_training.ipynb

@@ -313,11 +313,11 @@
    "outputs": [],
    "source": [
     "#@markdown ###Path to the training data:\n",
-    "training_source = \"./gdrive/MyDrive/CELLSEG_BENCHMARK/DATA/WNET/VIP_full\" #@param {type:\"string\"}\n",
-    "#@markdown ###Model name and path to model folder:\n",
-    "model_save_path = \"./gdrive/MyDrive/CELLSEG_BENCHMARK/WNET_TRAINING_RESULTS\" #@param {type:\"string\"}\n",
+    "training_source = \"./gdrive/MyDrive/path/to/data\" #@param {type:\"string\"}\n",
+    "#@markdown ###Path to save the weights (make sure to have enough space in your drive):\n",
+    "model_save_path = \"./gdrive/MyDrive/WNET_TRAINING_RESULTS\" #@param {type:\"string\"}\n",
     "#@markdown ---\n",
-    "#@markdown ###Perform validation on a test dataset\n",
+    "#@markdown ###Perform validation on a test dataset (optional):\n",
     "do_validation = False #@param {type:\"boolean\"}\n",
     "#@markdown ###Path to evaluation data (optional, use if checked above):\n",
     "eval_source = \"./gdrive/MyDrive/CELLSEG_BENCHMARK/DATA/WNET/eval/vol/\" #@param {type:\"string\"}\n",
@@ -396,13 +396,6 @@
     "        eval_label_folder,\n",
     "    ) if do_validation else None\n",
     "\n",
-    "try:\n",
-    "  import wandb\n",
-    "  WANDB_INSTALLED = True\n",
-    "except ImportError:\n",
-    "  WANDB_INSTALLED = False\n",
-    "\n",
-    "\n",
     "train_config = WNetTrainingWorkerConfig(\n",
     "    device=\"cuda:0\",\n",
     "    max_epochs=number_of_epochs,\n",