suayptalha
commited on
Commit
•
659ef62
1
Parent(s):
77e9390
Update configuration_minGRU.py
Browse files- configuration_minGRU.py +11 -11
configuration_minGRU.py
CHANGED
@@ -5,17 +5,17 @@ class MinGRUConfig(PretrainedConfig):
|
|
5 |
|
6 |
def __init__(
|
7 |
self,
|
8 |
-
vocab_size=30522,
|
9 |
-
d_model=512,
|
10 |
-
ff_mult=4,
|
11 |
-
expansion_factor=2,
|
12 |
-
depth=12,
|
13 |
-
n_layer=12,
|
14 |
-
pad_vocab_size_multiple=8,
|
15 |
-
initializer_range=0.02,
|
16 |
-
hidden_size=512,
|
17 |
-
hidden_dropout_prob=0.1,
|
18 |
-
num_labels=2,
|
19 |
**kwargs,
|
20 |
):
|
21 |
self.vocab_size = vocab_size
|
|
|
5 |
|
6 |
def __init__(
|
7 |
self,
|
8 |
+
vocab_size=30522,
|
9 |
+
d_model=512,
|
10 |
+
ff_mult=4,
|
11 |
+
expansion_factor=2,
|
12 |
+
depth=12,
|
13 |
+
n_layer=12,
|
14 |
+
pad_vocab_size_multiple=8,
|
15 |
+
initializer_range=0.02,
|
16 |
+
hidden_size=512,
|
17 |
+
hidden_dropout_prob=0.1,
|
18 |
+
num_labels=2,
|
19 |
**kwargs,
|
20 |
):
|
21 |
self.vocab_size = vocab_size
|