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_mbart.py
View File

@@ -219,7 +219,6 @@ class MBartModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase)
all_generative_model_classes = (MBartForConditionalGeneration,) if is_torch_available() else ()
is_encoder_decoder = True
test_pruning = False
test_head_masking = True
test_missing_keys = False
def setUp(self):