Update README.md
Browse files
README.md
CHANGED
@@ -166,6 +166,8 @@ We fine-tune `5CD-AI/visobert-14gb-corpus` on 4 downstream tasks with `transform
|
|
166 |
- training_epochs: 30
|
167 |
- model_max_length: 128
|
168 |
- learning_rate: 1e-5
|
|
|
|
|
169 |
|
170 |
And different additional configurations for each task:
|
171 |
| Emotion Recognition | Hate Speech Detection | Spam Reviews Detection | Hate Speech Spans Detection |
|
|
|
166 |
- training_epochs: 30
|
167 |
- model_max_length: 128
|
168 |
- learning_rate: 1e-5
|
169 |
+
- metric_for_best_model: wf1
|
170 |
+
- strategy: epoch
|
171 |
|
172 |
And different additional configurations for each task:
|
173 |
| Emotion Recognition | Hate Speech Detection | Spam Reviews Detection | Hate Speech Spans Detection |
|