Update README with proper number of epochs
Browse files
README.md
CHANGED
@@ -79,6 +79,6 @@ In fine-tuning, the following arguments were used:
|
|
79 |
|-------------------------------|-------|
|
80 |
| `per_device_train_batch_size` | 16 |
|
81 |
| `gradient_accumulation_steps` | 4 |
|
82 |
-
| `num_train_epochs` |
|
83 |
| `learning_rate` | 3e-4 |
|
84 |
| `warmup_steps` | 500 |
|
|
|
79 |
|-------------------------------|-------|
|
80 |
| `per_device_train_batch_size` | 16 |
|
81 |
| `gradient_accumulation_steps` | 4 |
|
82 |
+
| `num_train_epochs` | 20 |
|
83 |
| `learning_rate` | 3e-4 |
|
84 |
| `warmup_steps` | 500 |
|