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

Remove masked_lm_labels from returned dictionary (#7818)

Browse Source
This commit is contained in:
vblagoje
2020-10-16 03:12:10 -04:00
committed by GitHub
parent 96e47d9229
commit c6e865ac2b
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/data/data_collator.py
View File

@@ -449,8 +449,6 @@ class DataCollatorForNextSentencePrediction:
"labels": mlm_labels if self.mlm else None, "labels": mlm_labels if self.mlm else None,
"next_sentence_label": torch.tensor(nsp_labels), "next_sentence_label": torch.tensor(nsp_labels),
} }
if self.mlm:
result["masked_lm_labels"] = mlm_labels
return result return result
def _tensorize_batch(self, examples: List[torch.Tensor]) -> torch.Tensor: def _tensorize_batch(self, examples: List[torch.Tensor]) -> torch.Tensor: