elliotthwang
commited on
Commit
•
fd8a3ea
1
Parent(s):
de397ad
update model card README.md
Browse files
README.md
CHANGED
@@ -19,7 +19,7 @@ model-index:
|
|
19 |
metrics:
|
20 |
- name: Rouge1
|
21 |
type: rouge
|
22 |
-
value: 5.
|
23 |
---
|
24 |
|
25 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
@@ -30,10 +30,10 @@ should probably proofread and complete it, then remove this comment. -->
|
|
30 |
This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on the xlsum dataset.
|
31 |
It achieves the following results on the evaluation set:
|
32 |
- Loss: 2.9218
|
33 |
-
- Rouge1: 5.
|
34 |
-
- Rouge2: 1.
|
35 |
-
- Rougel: 5.
|
36 |
-
- Rougelsum: 5.
|
37 |
|
38 |
## Model description
|
39 |
|
@@ -64,17 +64,17 @@ The following hyperparameters were used during training:
|
|
64 |
|
65 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
66 |
|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
|
67 |
-
| 4.542 | 1.0 | 2336 | 3.1979 | 4.
|
68 |
-
| 3.7542 | 2.0 | 4672 | 3.0662 | 5.
|
69 |
-
| 3.5706 | 3.0 | 7008 | 3.0070 | 5.
|
70 |
-
| 3.4668 | 4.0 | 9344 | 2.9537 | 5.
|
71 |
-
| 3.4082 | 5.0 | 11680 | 2.9391 | 5.
|
72 |
-
| 3.375 | 6.0 | 14016 | 2.9218 | 5.
|
73 |
|
74 |
|
75 |
### Framework versions
|
76 |
|
77 |
- Transformers 4.20.1
|
78 |
-
- Pytorch 1.
|
79 |
- Datasets 2.3.2
|
80 |
- Tokenizers 0.12.1
|
|
|
19 |
metrics:
|
20 |
- name: Rouge1
|
21 |
type: rouge
|
22 |
+
value: 5.7806
|
23 |
---
|
24 |
|
25 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
|
|
30 |
This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on the xlsum dataset.
|
31 |
It achieves the following results on the evaluation set:
|
32 |
- Loss: 2.9218
|
33 |
+
- Rouge1: 5.7806
|
34 |
+
- Rouge2: 1.266
|
35 |
+
- Rougel: 5.761
|
36 |
+
- Rougelsum: 5.7833
|
37 |
|
38 |
## Model description
|
39 |
|
|
|
64 |
|
65 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
66 |
|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
|
67 |
+
| 4.542 | 1.0 | 2336 | 3.1979 | 4.8334 | 1.025 | 4.8142 | 4.8326 |
|
68 |
+
| 3.7542 | 2.0 | 4672 | 3.0662 | 5.2155 | 1.0978 | 5.2025 | 5.2158 |
|
69 |
+
| 3.5706 | 3.0 | 7008 | 3.0070 | 5.5471 | 1.3397 | 5.5386 | 5.5391 |
|
70 |
+
| 3.4668 | 4.0 | 9344 | 2.9537 | 5.5865 | 1.1558 | 5.5816 | 5.5964 |
|
71 |
+
| 3.4082 | 5.0 | 11680 | 2.9391 | 5.8061 | 1.3462 | 5.7944 | 5.812 |
|
72 |
+
| 3.375 | 6.0 | 14016 | 2.9218 | 5.7806 | 1.266 | 5.761 | 5.7833 |
|
73 |
|
74 |
|
75 |
### Framework versions
|
76 |
|
77 |
- Transformers 4.20.1
|
78 |
+
- Pytorch 1.12.0+cu113
|
79 |
- Datasets 2.3.2
|
80 |
- Tokenizers 0.12.1
|