hugojarkoff
commited on
Upload config.toml with huggingface_hub
Browse files- config.toml +5 -5
config.toml
CHANGED
@@ -9,18 +9,18 @@ split_train_val = 0.85
|
|
9 |
|
10 |
[model]
|
11 |
context_length = 256
|
12 |
-
num_heads =
|
13 |
-
num_layers =
|
14 |
-
hidden_dim =
|
15 |
|
16 |
[training]
|
17 |
num_training_steps = 20000
|
18 |
seed = 42
|
19 |
lr = 1e-3
|
20 |
-
batch_size =
|
21 |
|
22 |
[scheduler]
|
23 |
-
gamma = 0.
|
24 |
|
25 |
[evaluation]
|
26 |
sample_text = "Les vrais savent que "
|
|
|
9 |
|
10 |
[model]
|
11 |
context_length = 256
|
12 |
+
num_heads = 8
|
13 |
+
num_layers = 18
|
14 |
+
hidden_dim = 424
|
15 |
|
16 |
[training]
|
17 |
num_training_steps = 20000
|
18 |
seed = 42
|
19 |
lr = 1e-3
|
20 |
+
batch_size = 64
|
21 |
|
22 |
[scheduler]
|
23 |
+
gamma = 0.925
|
24 |
|
25 |
[evaluation]
|
26 |
sample_text = "Les vrais savent que "
|