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

Remove redundant test_head_masking = True flags in test files (#9858)

Browse Source 
* Remove redundant test_head_masking = True flags

* Remove all redundant test_head_masking flags in PyTorch test_modeling_* files

* Make test_head_masking = True as a default choice in test_modeling_tf_commong.py

* Remove all redundant test_head_masking flags in TensorFlow
test_modeling_tf_* files

* Put back test_head_masking=False fot TFT5 models
This commit is contained in:
Daniel Stancl
2021-01-28 16:09:13 +01:00
committed by GitHub
parent caddf9126b
commit 4c3ae89ad3
15 changed files with 1 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_tf_blenderbot.py
View File

@@ -177,7 +177,6 @@ class TFBlenderbotModelTest(TFModelTesterMixin, unittest.TestCase):
all_generative_model_classes = (TFBlenderbotForConditionalGeneration,) if is_tf_available() else ()
is_encoder_decoder = True
test_pruning = False
test_head_masking = True
def setUp(self):
self.model_tester = TFBlenderbotModelTester(self)