Model save
Browse files- README.md +16 -16
- generation_config.json +1 -1
README.md
CHANGED
@@ -17,11 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
|
|
17 |
|
18 |
This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on an unknown dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
-
- Loss: 0.
|
21 |
-
- Rouge1:
|
22 |
-
- Rouge2:
|
23 |
-
- Rougel:
|
24 |
-
- Rougelsum:
|
25 |
|
26 |
## Model description
|
27 |
|
@@ -52,21 +52,21 @@ The following hyperparameters were used during training:
|
|
52 |
|
53 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
54 |
|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
|
55 |
-
| 0.
|
56 |
-
| 0.
|
57 |
-
| 0.
|
58 |
-
| 0.
|
59 |
-
| 0.
|
60 |
-
| 0.
|
61 |
-
| 0.
|
62 |
-
| 0.
|
63 |
-
| 0.
|
64 |
-
| 0.
|
65 |
|
66 |
|
67 |
### Framework versions
|
68 |
|
69 |
-
- Transformers 4.
|
70 |
- Pytorch 2.3.1+cu121
|
71 |
- Datasets 2.20.0
|
72 |
- Tokenizers 0.19.1
|
|
|
17 |
|
18 |
This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on an unknown dataset.
|
19 |
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.1466
|
21 |
+
- Rouge1: 53.6452
|
22 |
+
- Rouge2: 39.2506
|
23 |
+
- Rougel: 44.3673
|
24 |
+
- Rougelsum: 44.3521
|
25 |
|
26 |
## Model description
|
27 |
|
|
|
52 |
|
53 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
54 |
|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
|
55 |
+
| 0.5077 | 1.0 | 36 | 0.2435 | 43.0788 | 24.6647 | 32.4588 | 32.5662 |
|
56 |
+
| 0.2391 | 2.0 | 72 | 0.2190 | 44.1416 | 26.995 | 33.4471 | 33.3775 |
|
57 |
+
| 0.2086 | 3.0 | 108 | 0.2094 | 47.6567 | 31.6759 | 37.2761 | 37.2626 |
|
58 |
+
| 0.1896 | 4.0 | 144 | 0.1986 | 49.8613 | 33.0333 | 38.6723 | 38.6662 |
|
59 |
+
| 0.1868 | 5.0 | 180 | 0.1945 | 49.5133 | 33.8958 | 38.6023 | 38.6402 |
|
60 |
+
| 0.1726 | 6.0 | 216 | 0.1745 | 52.5006 | 36.3466 | 41.5471 | 41.5067 |
|
61 |
+
| 0.1656 | 7.0 | 252 | 0.1705 | 52.6514 | 38.6746 | 43.3252 | 43.3525 |
|
62 |
+
| 0.1515 | 8.0 | 288 | 0.1580 | 53.659 | 37.2505 | 42.5857 | 42.6415 |
|
63 |
+
| 0.1561 | 9.0 | 324 | 0.1492 | 54.5155 | 39.0607 | 44.2114 | 44.3034 |
|
64 |
+
| 0.1363 | 10.0 | 360 | 0.1466 | 53.6452 | 39.2506 | 44.3673 | 44.3521 |
|
65 |
|
66 |
|
67 |
### Framework versions
|
68 |
|
69 |
+
- Transformers 4.43.2
|
70 |
- Pytorch 2.3.1+cu121
|
71 |
- Datasets 2.20.0
|
72 |
- Tokenizers 0.19.1
|
generation_config.json
CHANGED
@@ -11,6 +11,6 @@
|
|
11 |
"no_repeat_ngram_size": 3,
|
12 |
"num_beams": 4,
|
13 |
"pad_token_id": 1,
|
14 |
-
"transformers_version": "4.
|
15 |
"use_cache": false
|
16 |
}
|
|
|
11 |
"no_repeat_ngram_size": 3,
|
12 |
"num_beams": 4,
|
13 |
"pad_token_id": 1,
|
14 |
+
"transformers_version": "4.43.2",
|
15 |
"use_cache": false
|
16 |
}
|