Fix new unsup LR in tests

Author: C-Achard

napari_cellseg3d/_tests/test_unsup_training.py

@@ -5,15 +5,13 @@
     Trainer,
 )
 
+im_path = Path(__file__).resolve().parent / "res/test.tif"
 
-def test_unsupervised_worker(make_napari_viewer_proxy):
-    im_path = Path(__file__).resolve().parent / "res/test.tif"
-    # im_path_str = str(im_path)
 
+def test_unsupervised_worker(make_napari_viewer_proxy):
     unsup_viewer = make_napari_viewer_proxy()
     widget = Trainer(viewer=unsup_viewer)
     widget.device_choice.setCurrentIndex(0)
-
     widget.model_choice.setCurrentText("WNet")
     widget._toggle_unsupervised_mode(enabled=True)
 

napari_cellseg3d/config.py

@@ -285,6 +285,7 @@ class WNetTrainingWorkerConfig(TrainingWorkerConfig):
     out_channels: int = 1  # decoder (reconstruction) output channels
     num_classes: int = 2  # encoder output channels
     dropout: float = 0.65
+    learning_rate: np.float64 = 2e-5
     use_clipping: bool = False  # use gradient clipping
     clipping: float = 1.0  # clipping value
     weight_decay: float = 0.01  # 1e-5  # weight decay (used 0.01 historically)