Update README.md
Browse files
README.md
CHANGED
|
@@ -177,7 +177,7 @@ mask_filler("shop làm ăn như cái <mask>", top_k=10)
|
|
| 177 |
We fine-tune `5CD-AI/visocial-T5-base` on 3 downstream tasks with `transformers` library with the following configuration:
|
| 178 |
- seed: 42
|
| 179 |
- training_epochs: 4
|
| 180 |
-
- train_batch_size: 4
|
| 181 |
- gradient_accumulation_steps: 8
|
| 182 |
- learning_rate: 3e-4
|
| 183 |
- lr_scheduler_type: linear
|
|
|
|
| 177 |
We fine-tune `5CD-AI/visocial-T5-base` on 3 downstream tasks with `transformers` library with the following configuration:
|
| 178 |
- seed: 42
|
| 179 |
- training_epochs: 4
|
| 180 |
+
- train_batch_size: 4
|
| 181 |
- gradient_accumulation_steps: 8
|
| 182 |
- learning_rate: 3e-4
|
| 183 |
- lr_scheduler_type: linear
|