Test disabling smoothing

Author: C-Achard

napari_cellseg3d/code_models/worker_training.py

@@ -132,12 +132,12 @@ def __init__(
         self.downloader = WeightsDownloader()
 
         self.loss_dict = {
-            "Dice": DiceLoss(sigmoid=True),
+            "Dice": DiceLoss(sigmoid=True, smooth_nr=0.0),
             # "BCELoss": torch.nn.BCELoss(), # dev
             # "BCELogits": torch.nn.BCEWithLogitsLoss(),
             "Generalized Dice": GeneralizedDiceLoss(sigmoid=True),
             "DiceCE": DiceCELoss(sigmoid=True, lambda_ce=0.5),
-            "Tversky": TverskyLoss(sigmoid=True),
+            "Tversky": TverskyLoss(sigmoid=True, smooth_nr=0.0),
             # "Focal loss": FocalLoss(),
             # "Dice-Focal loss": DiceFocalLoss(sigmoid=True, lambda_dice=0.5),
         }