6a1fbe59415422ee0cbaa6d5f5e21a76
This model is a fine-tuned version of distilbert/distilbert-base-german-cased on the nyu-mll/glue [sst2] dataset. It achieves the following results on the evaluation set:
- Loss: 0.5776
- Data Size: 1.0
- Epoch Runtime: 54.9319
- Accuracy: 0.8565
- F1 Macro: 0.8555
- Rouge1: 0.8565
- Rouge2: 0.0
- Rougel: 0.8565
- Rougelsum: 0.8565
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.6997 | 0 | 0.8164 | 0.4907 | 0.3292 | 0.4907 | 0.0 | 0.4907 | 0.4919 |
| No log | 1 | 2104 | 0.7071 | 0.0078 | 1.4184 | 0.5116 | 0.3467 | 0.5116 | 0.0 | 0.5116 | 0.5104 |
| No log | 2 | 4208 | 0.6845 | 0.0156 | 1.8154 | 0.5150 | 0.3544 | 0.5150 | 0.0 | 0.5150 | 0.5150 |
| 0.0143 | 3 | 6312 | 0.5536 | 0.0312 | 2.7219 | 0.7407 | 0.7407 | 0.7407 | 0.0 | 0.7407 | 0.7407 |
| 0.5296 | 4 | 8416 | 0.4822 | 0.0625 | 4.4472 | 0.7639 | 0.7613 | 0.7639 | 0.0 | 0.7639 | 0.7639 |
| 0.4061 | 5 | 10520 | 0.4130 | 0.125 | 7.8497 | 0.8113 | 0.8112 | 0.8113 | 0.0 | 0.8125 | 0.8125 |
| 0.305 | 6 | 12624 | 0.4246 | 0.25 | 14.6031 | 0.8252 | 0.8233 | 0.8252 | 0.0 | 0.8252 | 0.8252 |
| 0.2494 | 7 | 14728 | 0.3525 | 0.5 | 28.1423 | 0.8600 | 0.8597 | 0.8600 | 0.0 | 0.8600 | 0.8600 |
| 0.1801 | 8.0 | 16832 | 0.3715 | 1.0 | 54.5189 | 0.8507 | 0.8502 | 0.8519 | 0.0 | 0.8507 | 0.8507 |
| 0.1387 | 9.0 | 18936 | 0.4486 | 1.0 | 54.7929 | 0.8588 | 0.8587 | 0.8588 | 0.0 | 0.8588 | 0.8588 |
| 0.1142 | 10.0 | 21040 | 0.5350 | 1.0 | 55.5843 | 0.8576 | 0.8573 | 0.8576 | 0.0 | 0.8576 | 0.8588 |
| 0.0993 | 11.0 | 23144 | 0.5776 | 1.0 | 54.9319 | 0.8565 | 0.8555 | 0.8565 | 0.0 | 0.8565 | 0.8565 |
Framework versions
- Transformers 4.57.0
- Pytorch 2.8.0+cu128
- Datasets 4.3.0
- Tokenizers 0.22.1
- Downloads last month
- 2
Model tree for contemmcm/6a1fbe59415422ee0cbaa6d5f5e21a76
Base model
distilbert/distilbert-base-german-cased