Headmasking

2019-11-07 17:29:29 +00:00
parent abb23a78ba
commit 16263f9685
2 changed files with 6 additions and 6 deletions
--- a/transformers/tests/modeling_albert_test.py
+++ b/transformers/tests/modeling_albert_test.py
@@ -35,7 +35,6 @@ else:
 class AlbertModelTest(CommonTestCases.CommonModelTester):

    all_model_classes = (AlbertModel, AlbertForMaskedLM) if is_torch_available() else ()
-    test_head_masking = False

    class AlbertModelTester(object):