Update configuration_minGRULM.py
Browse files
configuration_minGRULM.py
CHANGED
@@ -14,6 +14,7 @@ class MinGRULMConfig(PretrainedConfig):
|
|
14 |
n_layer=12,
|
15 |
pad_vocab_size_multiple=8,
|
16 |
initializer_range=0.02,
|
|
|
17 |
**kwargs,
|
18 |
):
|
19 |
self.vocab_size = vocab_size
|
@@ -23,6 +24,7 @@ class MinGRULMConfig(PretrainedConfig):
|
|
23 |
self.expand = expand
|
24 |
self.depth = depth
|
25 |
self.n_layer = n_layer
|
|
|
26 |
self.pad_vocab_size_multiple = pad_vocab_size_multiple
|
27 |
self.initializer_range = initializer_range
|
28 |
super().__init__(**kwargs)
|
|
|
14 |
n_layer=12,
|
15 |
pad_vocab_size_multiple=8,
|
16 |
initializer_range=0.02,
|
17 |
+
hidden_size = 512,
|
18 |
**kwargs,
|
19 |
):
|
20 |
self.vocab_size = vocab_size
|
|
|
24 |
self.expand = expand
|
25 |
self.depth = depth
|
26 |
self.n_layer = n_layer
|
27 |
+
self.hidden_size = hidden_size
|
28 |
self.pad_vocab_size_multiple = pad_vocab_size_multiple
|
29 |
self.initializer_range = initializer_range
|
30 |
super().__init__(**kwargs)
|