SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fixed TransfoXLLMHeadModel documentation (#3661)

Browse Source 
Co-authored-by: TevenLeScao <teven.lescao@gmail.com>
This commit is contained in:
Teven
2020-04-07 00:47:51 +02:00
committed by GitHub
parent 96ab75b8dd
commit 0a9d09b42a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/modeling_transfo_xl.py
View File

@@ -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).