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 "
|