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

Fix label name in DataCollatorForNextSentencePrediction test (#8048)

Browse Source
This commit is contained in:
Sylvain Gugger
2020-10-26 09:23:12 -04:00
committed by GitHub
parent 8bbe8247f1
commit 077478637d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_data_collator.py
View File

@@ -175,7 +175,7 @@ class DataCollatorIntegrationTest(unittest.TestCase):
total_samples = batch["input_ids"].shape[0]
self.assertEqual(batch["input_ids"].shape, torch.Size((total_samples, 512)))
self.assertEqual(batch["token_type_ids"].shape, torch.Size((total_samples, 512)))
self.assertEqual(batch["masked_lm_labels"].shape, torch.Size((total_samples, 512)))
self.assertEqual(batch["labels"].shape, torch.Size((total_samples, 512)))
self.assertEqual(batch["next_sentence_label"].shape, torch.Size((total_samples,)))
@slow