Update README.md
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ overwrite_output_dir: false
|
|
| 59 |
|
| 60 |
### train
|
| 61 |
per_device_train_batch_size: 2
|
| 62 |
-
gradient_accumulation_steps:
|
| 63 |
learning_rate: 0.00001
|
| 64 |
num_train_epochs: 1
|
| 65 |
lr_scheduler_type: cosine
|
|
|
|
| 59 |
|
| 60 |
### train
|
| 61 |
per_device_train_batch_size: 2
|
| 62 |
+
gradient_accumulation_steps: 4
|
| 63 |
learning_rate: 0.00001
|
| 64 |
num_train_epochs: 1
|
| 65 |
lr_scheduler_type: cosine
|