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

Merge branch 'master' into improved_testing

Browse Source
This commit is contained in:
Thomas Wolf
2019-08-30 13:40:35 +02:00
committed by GitHub
parent f8aace6bcd f7978490b2
commit 50e615f43d
43 changed files with 3103 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
pytorch_transformers/modeling_gpt2.py
View File

@@ -657,9 +657,6 @@ class GPT2DoubleHeadsModel(GPT2PreTrainedModel):
Indices should be in ``[0, ..., num_choices]`` where `num_choices` is the size of the second dimension
of the input tensors. (see `input_ids` above)
`multiple_choice_labels`: optional multiple choice labels: ``torch.LongTensor`` of shape [batch_size]
with indices selected in [0, ..., num_choices].
Outputs: `Tuple` comprising various elements depending on the configuration (config) and inputs:
**lm_loss**: (`optional`, returned when ``lm_labels`` is provided) ``torch.FloatTensor`` of shape ``(1,)``:
Language modeling loss.