97ba92b93f09f714f7ad1181c96d6ee9
This model is a fine-tuned version of distilbert/distilbert-base-multilingual-cased on the nyu-mll/glue [qnli] dataset. It achieves the following results on the evaluation set:
- Loss: 0.5529
- Data Size: 1.0
- Epoch Runtime: 99.5191
- Accuracy: 0.8438
- F1 Macro: 0.8437
- Rouge1: 0.8438
- Rouge2: 0.0
- Rougel: 0.8436
- Rougelsum: 0.8439
Model description
More information needed
Intended uses & limitations
More information needed
Training and evaluation data
More information needed
Training procedure
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- distributed_type: multi-GPU
- num_devices: 4
- total_train_batch_size: 32
- total_eval_batch_size: 32
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
- lr_scheduler_type: constant
- num_epochs: 50
Training results
| Training Loss | Epoch | Step | Validation Loss | Data Size | Epoch Runtime | Accuracy | F1 Macro | Rouge1 | Rouge2 | Rougel | Rougelsum |
|---|---|---|---|---|---|---|---|---|---|---|---|
| No log | 0 | 0 | 0.6978 | 0 | 2.0292 | 0.4921 | 0.3333 | 0.4926 | 0.0 | 0.4923 | 0.4923 |
| No log | 1 | 3273 | 0.6034 | 0.0078 | 3.0278 | 0.7103 | 0.7056 | 0.7099 | 0.0 | 0.7101 | 0.7099 |
| 0.0101 | 2 | 6546 | 0.5186 | 0.0156 | 3.7742 | 0.7586 | 0.7518 | 0.7588 | 0.0 | 0.7586 | 0.7585 |
| 0.5081 | 3 | 9819 | 0.4626 | 0.0312 | 5.6026 | 0.7930 | 0.7929 | 0.7932 | 0.0 | 0.7932 | 0.7930 |
| 0.4966 | 4 | 13092 | 0.4292 | 0.0625 | 8.6660 | 0.8085 | 0.8084 | 0.8083 | 0.0 | 0.8086 | 0.8083 |
| 0.4436 | 5 | 16365 | 0.4058 | 0.125 | 15.3781 | 0.8224 | 0.8224 | 0.8224 | 0.0 | 0.8222 | 0.8222 |
| 0.4434 | 6 | 19638 | 0.4361 | 0.25 | 27.3107 | 0.8127 | 0.8112 | 0.8127 | 0.0 | 0.8129 | 0.8127 |
| 0.3919 | 7 | 22911 | 0.3749 | 0.5 | 51.4249 | 0.8465 | 0.8464 | 0.8465 | 0.0 | 0.8467 | 0.8465 |
| 0.3421 | 8.0 | 26184 | 0.3587 | 1.0 | 99.6412 | 0.8465 | 0.8462 | 0.8467 | 0.0 | 0.8467 | 0.8467 |
| 0.2472 | 9.0 | 29457 | 0.3837 | 1.0 | 100.2470 | 0.8493 | 0.8492 | 0.8492 | 0.0 | 0.8491 | 0.8494 |
| 0.1704 | 10.0 | 32730 | 0.4761 | 1.0 | 100.1392 | 0.8471 | 0.8469 | 0.8471 | 0.0 | 0.8472 | 0.8471 |
| 0.158 | 11.0 | 36003 | 0.4926 | 1.0 | 101.5143 | 0.8388 | 0.8386 | 0.8388 | 0.0 | 0.8390 | 0.8390 |
| 0.1273 | 12.0 | 39276 | 0.5529 | 1.0 | 99.5191 | 0.8438 | 0.8437 | 0.8438 | 0.0 | 0.8436 | 0.8439 |
Framework versions
- Transformers 4.57.0
- Pytorch 2.8.0+cu128
- Datasets 4.3.0
- Tokenizers 0.22.1
- Downloads last month
- 1