update model card README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,8 @@ license: apache-2.0
|
|
| 3 |
tags:
|
| 4 |
- summarization
|
| 5 |
- generated_from_trainer
|
|
|
|
|
|
|
| 6 |
model-index:
|
| 7 |
- name: t5-small-train
|
| 8 |
results: []
|
|
@@ -15,15 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
-
-
|
| 19 |
-
-
|
| 20 |
-
-
|
| 21 |
-
-
|
| 22 |
-
-
|
| 23 |
-
- eval_runtime: 11.6689
|
| 24 |
-
- eval_samples_per_second: 13.026
|
| 25 |
-
- eval_steps_per_second: 1.371
|
| 26 |
-
- step: 0
|
| 27 |
|
| 28 |
## Model description
|
| 29 |
|
|
@@ -43,13 +41,29 @@ More information needed
|
|
| 43 |
|
| 44 |
The following hyperparameters were used during training:
|
| 45 |
- learning_rate: 5.6e-05
|
| 46 |
-
- train_batch_size:
|
| 47 |
-
- eval_batch_size:
|
| 48 |
- seed: 42
|
| 49 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 50 |
- lr_scheduler_type: linear
|
| 51 |
- num_epochs: 10
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
### Framework versions
|
| 54 |
|
| 55 |
- Transformers 4.18.0
|
|
|
|
| 3 |
tags:
|
| 4 |
- summarization
|
| 5 |
- generated_from_trainer
|
| 6 |
+
metrics:
|
| 7 |
+
- rouge
|
| 8 |
model-index:
|
| 9 |
- name: t5-small-train
|
| 10 |
results: []
|
|
|
|
| 17 |
|
| 18 |
This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
|
| 19 |
It achieves the following results on the evaluation set:
|
| 20 |
+
- Loss: 2.4110
|
| 21 |
+
- Rouge1: 41.006
|
| 22 |
+
- Rouge2: 18.9406
|
| 23 |
+
- Rougel: 35.7319
|
| 24 |
+
- Rougelsum: 35.9987
|
|
|
|
|
|
|
|
|
|
|
|
|
| 25 |
|
| 26 |
## Model description
|
| 27 |
|
|
|
|
| 41 |
|
| 42 |
The following hyperparameters were used during training:
|
| 43 |
- learning_rate: 5.6e-05
|
| 44 |
+
- train_batch_size: 8
|
| 45 |
+
- eval_batch_size: 8
|
| 46 |
- seed: 42
|
| 47 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 48 |
- lr_scheduler_type: linear
|
| 49 |
- num_epochs: 10
|
| 50 |
|
| 51 |
+
### Training results
|
| 52 |
+
|
| 53 |
+
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
| 54 |
+
|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
|
| 55 |
+
| 2.5097 | 1.0 | 45 | 2.5765 | 36.9095 | 15.7531 | 32.4588 | 32.7501 |
|
| 56 |
+
| 2.39 | 2.0 | 90 | 2.4823 | 39.1984 | 17.2602 | 34.5018 | 34.8303 |
|
| 57 |
+
| 2.2862 | 3.0 | 135 | 2.4521 | 39.9179 | 18.2643 | 35.4775 | 35.7854 |
|
| 58 |
+
| 2.2011 | 4.0 | 180 | 2.4314 | 40.1014 | 18.3646 | 35.274 | 35.5883 |
|
| 59 |
+
| 2.1335 | 5.0 | 225 | 2.4240 | 40.1053 | 18.406 | 35.0905 | 35.3427 |
|
| 60 |
+
| 2.0803 | 6.0 | 270 | 2.4178 | 41.1202 | 18.5746 | 35.5454 | 35.7857 |
|
| 61 |
+
| 2.0662 | 7.0 | 315 | 2.4129 | 40.7965 | 18.5148 | 35.5866 | 35.8591 |
|
| 62 |
+
| 2.0291 | 8.0 | 360 | 2.4103 | 40.7121 | 18.8736 | 35.6646 | 35.9392 |
|
| 63 |
+
| 1.9807 | 9.0 | 405 | 2.4112 | 40.9464 | 18.9815 | 35.8468 | 36.1114 |
|
| 64 |
+
| 1.9702 | 10.0 | 450 | 2.4110 | 41.006 | 18.9406 | 35.7319 | 35.9987 |
|
| 65 |
+
|
| 66 |
+
|
| 67 |
### Framework versions
|
| 68 |
|
| 69 |
- Transformers 4.18.0
|