From a63f7195ff8b0960c65e715b3d85edea037b16c0 Mon Sep 17 00:00:00 2001 From: Babak Azad Date: Wed, 18 Aug 2021 23:11:14 +0200 Subject: [PATCH] Parameter description added Co-authored-by: Armand <83031821+hermancollin@users.noreply.github.com> --- ivadomed/losses.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ivadomed/losses.py b/ivadomed/losses.py index fd400e2c1..241e7ffaa 100644 --- a/ivadomed/losses.py +++ b/ivadomed/losses.py @@ -50,6 +50,7 @@ class DiceLoss(nn.Module): Attributes: smooth (float): Value to avoid division by zero when images and predictions are empty. + sample_wise (bool): If set, dice loss is averaged over sample instead of over batch """ def __init__(self, smooth=1.0, sample_wise=False): super(DiceLoss, self).__init__()