Update README.md
Browse files
README.md
CHANGED
|
@@ -50,14 +50,23 @@ The model was evaluated on test dataset. Below are key metrics:
|
|
| 50 |
**Test Dataset Description:** size ~94 900 text rows
|
| 51 |
|
| 52 |
# Used Hyperparameters
|
|
|
|
| 53 |
**Evaluation strategy:** epoch
|
|
|
|
| 54 |
**Train batch size per device:** 16
|
|
|
|
| 55 |
**Learning rate:** 1e-5
|
|
|
|
| 56 |
**Scheduler:** linear
|
|
|
|
| 57 |
**Warmup steps:** 500
|
|
|
|
| 58 |
**Optimizer:** AdamW
|
|
|
|
| 59 |
**Number of epochs:** 14
|
|
|
|
| 60 |
**FP16 mixed precision training:** True
|
|
|
|
| 61 |
**Half precision backend:** cuda_amp
|
| 62 |
|
| 63 |
|
|
|
|
| 50 |
**Test Dataset Description:** size ~94 900 text rows
|
| 51 |
|
| 52 |
# Used Hyperparameters
|
| 53 |
+
|
| 54 |
**Evaluation strategy:** epoch
|
| 55 |
+
|
| 56 |
**Train batch size per device:** 16
|
| 57 |
+
|
| 58 |
**Learning rate:** 1e-5
|
| 59 |
+
|
| 60 |
**Scheduler:** linear
|
| 61 |
+
|
| 62 |
**Warmup steps:** 500
|
| 63 |
+
|
| 64 |
**Optimizer:** AdamW
|
| 65 |
+
|
| 66 |
**Number of epochs:** 14
|
| 67 |
+
|
| 68 |
**FP16 mixed precision training:** True
|
| 69 |
+
|
| 70 |
**Half precision backend:** cuda_amp
|
| 71 |
|
| 72 |
|