Upload best_model_config.json with huggingface_hub
Browse files- best_model_config.json +20 -0
best_model_config.json
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"vocab_size": 1368,
|
| 3 |
+
"d_model": 1536,
|
| 4 |
+
"n_heads": 24,
|
| 5 |
+
"n_layers": 28,
|
| 6 |
+
"d_ff": 6144,
|
| 7 |
+
"max_seq_len": 400,
|
| 8 |
+
"dropout": 0.1,
|
| 9 |
+
"bos_token": 0,
|
| 10 |
+
"eos_token": 1,
|
| 11 |
+
"pad_token": 2,
|
| 12 |
+
"phrase_end_token": 3,
|
| 13 |
+
"hold_token": 4,
|
| 14 |
+
"learning_rate": 0.0001,
|
| 15 |
+
"weight_decay": 0.01,
|
| 16 |
+
"warmup_steps": 2000,
|
| 17 |
+
"gradient_clip_val": 0.5,
|
| 18 |
+
"epoch": 8,
|
| 19 |
+
"val_loss": 0.7097084007479928
|
| 20 |
+
}
|