Skip to content

GitLab

Unverified Commit c6e865ac authored by vblagoje's avatar vblagoje Committed by GitHub
Browse files

Remove masked_lm_labels from returned dictionary (#7818)

parent 96e47d92
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
src/transformers/data/data_collator.py
View file @ c6e865ac
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment