Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ For best results, format all inputs in this manner. **Make sure to include a new
|
|
37 |
The following hyperparameters were used during DPO training:
|
38 |
- learning_rate: 1e-6 * sqrt(Num of Nodes)
|
39 |
- total_train_batch_size: 128 * Num of Nodes
|
40 |
-
- optimizer: AdamW with beta1 0.9, beta2 0.999 and epsilon 1e-
|
41 |
- lr_scheduler_type: linear
|
42 |
- lr_scheduler_warmup_ratio: 0.1
|
43 |
- Weight Decay: 0.05
|
|
|
37 |
The following hyperparameters were used during DPO training:
|
38 |
- learning_rate: 1e-6 * sqrt(Num of Nodes)
|
39 |
- total_train_batch_size: 128 * Num of Nodes
|
40 |
+
- optimizer: AdamW with beta1 0.9, beta2 0.999 and epsilon 1e-8
|
41 |
- lr_scheduler_type: linear
|
42 |
- lr_scheduler_warmup_ratio: 0.1
|
43 |
- Weight Decay: 0.05
|