From 35d79c1aa5fb26ba0b2c1ec67084c66d43687220 Mon Sep 17 00:00:00 2001 From: Caleb Robinson Date: Fri, 17 Sep 2021 23:59:02 -0700 Subject: [PATCH] Update docstring for the JaccardLoss (#487) JaccardLoss doesn't accept a parameter "ignore_index" --- segmentation_models_pytorch/losses/jaccard.py | 1 - 1 file changed, 1 deletion(-) diff --git a/segmentation_models_pytorch/losses/jaccard.py b/segmentation_models_pytorch/losses/jaccard.py index 33b776cf..5e2c0b06 100644 --- a/segmentation_models_pytorch/losses/jaccard.py +++ b/segmentation_models_pytorch/losses/jaccard.py @@ -29,7 +29,6 @@ def __init__( log_loss: If True, loss computed as `- log(jaccard_coeff)`, otherwise `1 - jaccard_coeff` from_logits: If True, assumes input is raw logits smooth: Smoothness constant for dice coefficient - ignore_index: Label that indicates ignored pixels (does not contribute to loss) eps: A small epsilon for numerical stability to avoid zero division error (denominator will be always greater or equal to eps)