Update README.md
Browse files
README.md
CHANGED
|
@@ -34,6 +34,10 @@ More information needed
|
|
| 34 |
|
| 35 |
### Training hyperparameters
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
The following hyperparameters were used during training:
|
| 38 |
- learning_rate: 0.0002
|
| 39 |
- train_batch_size: 2
|
|
|
|
| 34 |
|
| 35 |
### Training hyperparameters
|
| 36 |
|
| 37 |
+
lora_alpha = 32
|
| 38 |
+
lora_dropout = 0.05
|
| 39 |
+
lora_rank = 16
|
| 40 |
+
|
| 41 |
The following hyperparameters were used during training:
|
| 42 |
- learning_rate: 0.0002
|
| 43 |
- train_batch_size: 2
|