This commit is contained in:
LysandreJik committed 2019-08-08 10:38:57 -04:00
1 parent 7df303f5ad
commit f2b300df6b
3 files changed
+4 -4

No files matched your search

+1 -1
View File
@@ -784,7 +784,7 @@ class XLMForSequenceClassification(XLMPreTrainedModel):
r"""
**labels**: (`optional`) ``torch.LongTensor`` of shape ``(batch_size,)``:
Labels for computing the sequence classification/regression loss.
Indices should be in ``[0, ..., config.num_labels]``.
Indices should be in ``[0, ..., config.num_labels - 1]``.
If ``config.num_labels == 1`` a regression loss is computed (Mean-Square loss),
If ``config.num_labels > 1`` a classification loss is computed (Cross-Entropy).