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

@@ -75,6 +75,7 @@ class TFModelTesterMixin:
all_model_classes = ()
all_generative_model_classes = ()
test_resize_embeddings = True
test_head_masking = True
is_encoder_decoder = False
def _prepare_for_class(self, inputs_dict, model_class, return_labels=False) -> dict: