From c6e865ac2bdad9f65c5bc51561563c46c427b506 Mon Sep 17 00:00:00 2001
From: vblagoje <dovlex@gmail.com>
Date: Fri, 16 Oct 2020 03:12:10 -0400
Subject: [PATCH] Remove masked_lm_labels from returned dictionary (#7818)

---
 src/transformers/data/data_collator.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/src/transformers/data/data_collator.py b/src/transformers/data/data_collator.py
index 4cf1b8919c..e46bba13e3 100644
--- a/src/transformers/data/data_collator.py
+++ b/src/transformers/data/data_collator.py
@@ -449,8 +449,6 @@ class DataCollatorForNextSentencePrediction:
             "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: