From 55bcae7f25678bd8f40cda1a076ef9773047c1fe Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Tue, 31 Mar 2020 15:32:25 +0200 Subject: [PATCH] remove useless and confusing lm_labels line (#3531) --- src/transformers/modeling_t5.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/transformers/modeling_t5.py b/src/transformers/modeling_t5.py index af064c2ace..98fc319a4e 100644 --- a/src/transformers/modeling_t5.py +++ b/src/transformers/modeling_t5.py @@ -901,7 +901,6 @@ class T5ForConditionalGeneration(T5PreTrainedModel): lm_labels (:obj:`torch.LongTensor` of shape :obj:`(batch_size,)`, `optional`, defaults to :obj:`None`): Labels for computing the sequence classification/regression loss. Indices should be in :obj:`[-100, 0, ..., config.vocab_size - 1]`. - If :obj:`config.num_labels > 1` a classification loss is computed (Cross-Entropy). All labels set to ``-100`` are ignored (masked), the loss is only computed for labels in ``[0, ..., config.vocab_size]``