Skip to content

Commit 35d79c1

Browse files
authored
Update docstring for the JaccardLoss (#487)
JaccardLoss doesn't accept a parameter "ignore_index"
1 parent 15418da commit 35d79c1

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

segmentation_models_pytorch/losses/jaccard.py

-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ def __init__(
2929
log_loss: If True, loss computed as `- log(jaccard_coeff)`, otherwise `1 - jaccard_coeff`
3030
from_logits: If True, assumes input is raw logits
3131
smooth: Smoothness constant for dice coefficient
32-
ignore_index: Label that indicates ignored pixels (does not contribute to loss)
3332
eps: A small epsilon for numerical stability to avoid zero division error
3433
(denominator will be always greater or equal to eps)
3534

0 commit comments

Comments
 (0)