Fix input embeddings

2019-11-26 12:59:28 -05:00
parent bdfe21ab24
commit f2f329408d
3 changed files with 8 additions and 3 deletions
--- a/transformers/tests/modeling_albert_test.py
+++ b/transformers/tests/modeling_albert_test.py
@@ -49,6 +49,7 @@ class AlbertModelTest(CommonTestCases.CommonModelTester):
                     use_token_type_ids=True,
                     use_labels=True,
                     vocab_size=99,
+                     embedding_size=16,
                     hidden_size=36,
                     num_hidden_layers=6,
                     num_hidden_groups=6,
@@ -73,6 +74,7 @@ class AlbertModelTest(CommonTestCases.CommonModelTester):
            self.use_token_type_ids = use_token_type_ids
            self.use_labels = use_labels
            self.vocab_size = vocab_size
+            self.embedding_size = embedding_size
            self.hidden_size = hidden_size
            self.num_hidden_layers = num_hidden_layers
            self.num_attention_heads = num_attention_heads