Remove masked_lm_labels from returned dictionary (huggingface#7818)

bratao · Oct 16, 2020 · c6e865a · c6e865a
1 parent 96e47d9
commit c6e865a
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py
@@ -449,8 +449,6 @@ def __call__(self, examples: List[Dict[str, torch.Tensor]]) -> Dict[str, torch.T
             "labels": mlm_labels if self.mlm else None,
             "next_sentence_label": torch.tensor(nsp_labels),
         }
-        if self.mlm:
-            result["masked_lm_labels"] = mlm_labels
         return result
 
     def _tensorize_batch(self, examples: List[torch.Tensor]) -> torch.Tensor: