From 0a9d09b42a9c7c1ccc00da48486a1188078e8594 Mon Sep 17 00:00:00 2001 From: Teven Date: Tue, 7 Apr 2020 00:47:51 +0200 Subject: [PATCH] fixed TransfoXLLMHeadModel documentation (#3661) Co-authored-by: TevenLeScao --- src/transformers/modeling_transfo_xl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/modeling_transfo_xl.py b/src/transformers/modeling_transfo_xl.py index c3bd4076b1..1ed8408966 100644 --- a/src/transformers/modeling_transfo_xl.py +++ b/src/transformers/modeling_transfo_xl.py @@ -859,7 +859,7 @@ class TransfoXLLMHeadModel(TransfoXLPreTrainedModel): Return: :obj:`tuple(torch.FloatTensor)` comprising various elements depending on the configuration (:class:`~transformers.TransfoXLConfig`) and inputs: - loss (:obj:`torch.FloatTensor` of shape `(1,)`, `optional`, returned when ``labels`` is provided) + loss (:obj:`torch.FloatTensor` of shape `(batch_size, sequence_length)`, `optional`, returned when ``labels`` is provided) Language modeling loss. prediction_scores (:obj:`torch.FloatTensor` of shape :obj:`(batch_size, sequence_length, config.vocab_size)`): Prediction scores of the language modeling head (scores for each vocabulary token before SoftMax).