diff --git a/torch/nn/modules/loss.py b/torch/nn/modules/loss.py index 81e2335933f1b..e5a3f44cf660e 100644 --- a/torch/nn/modules/loss.py +++ b/torch/nn/modules/loss.py @@ -467,7 +467,7 @@ class MultiLabelMarginLoss(_Loss): `y` and `x` must have the same size. - The criterion only considers the first non zero `y[j]` targets. + The criterion only considers the first non-negative `y[j]` targets. This allows for different samples to have variable amounts of target classes """