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

Switch return_dict to True by default. (#8530)

Browse Source 
* Use the CI to identify failing tests

* Remove from all examples and tests

* More default switch

* Fixes

* More test fixes

* More fixes

* Last fixes hopefully

* Use the CI to identify failing tests

* Remove from all examples and tests

* More default switch

* Fixes

* More test fixes

* More fixes

* Last fixes hopefully

* Run on the real suite

* Fix slow tests
This commit is contained in:
Sylvain Gugger
2020-11-16 11:43:00 -05:00
committed by GitHub
parent 0d0a0785fd
commit 1073a2bde5
106 changed files with 138 additions and 234 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_modeling_lxmert.py
View File

@@ -282,7 +282,6 @@ class LxmertModelTester:
attention_mask=input_mask,
labels=ans,
output_attentions=output_attentions,
return_dict=True,
)
result = model(input_ids, visual_feats, bounding_boxes, labels=ans)
result = model(
@@ -302,7 +301,6 @@ class LxmertModelTester:
attention_mask=input_mask,
labels=ans,
output_attentions=not output_attentions,
return_dict=True,
)
self.parent.assertEqual(result.question_answering_score.shape, (self.batch_size, self.num_qa_labels))
@@ -335,7 +333,6 @@ class LxmertModelTester:
matched_label=matched_label,
ans=ans,
output_attentions=output_attentions,
return_dict=True,
)
result = model(
input_ids,
@@ -390,7 +387,6 @@ class LxmertModelTester:
matched_label=matched_label,
ans=ans,
output_attentions=not output_attentions,
return_dict=True,
)
self.parent.assertEqual(result.prediction_logits.shape, (self.batch_size, self.seq_length, self.vocab_size))
@@ -427,7 +423,6 @@ class LxmertModelTester:
token_type_ids=token_type_ids,
attention_mask=input_mask,
ans=ans,
return_dict=True,
)
result_qa = model_qa(
@@ -437,7 +432,6 @@ class LxmertModelTester:
labels=ans,
token_type_ids=token_type_ids,
attention_mask=input_mask,
return_dict=True,
)
model_pretrain.resize_num_qa_labels(num_small_labels)
@@ -450,7 +444,6 @@ class LxmertModelTester:
token_type_ids=token_type_ids,
attention_mask=input_mask,
ans=less_labels_ans,
return_dict=True,
)
result_qa_less = model_qa(
@@ -460,7 +453,6 @@ class LxmertModelTester:
labels=less_labels_ans,
token_type_ids=token_type_ids,
attention_mask=input_mask,
return_dict=True,
)
model_pretrain.resize_num_qa_labels(num_large_labels)
@@ -473,7 +465,6 @@ class LxmertModelTester:
token_type_ids=token_type_ids,
attention_mask=input_mask,
ans=more_labels_ans,
return_dict=True,
)
result_qa_more = model_qa(
@@ -483,7 +474,6 @@ class LxmertModelTester:
labels=more_labels_ans,
token_type_ids=token_type_ids,
attention_mask=input_mask,
return_dict=True,
)
model_qa_labels = model_qa.num_qa_labels