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