Training complete!
Browse files- README.md +3 -10
- generation_config.json +1 -1
- tokenizer.json +14 -2
README.md
CHANGED
@@ -2,8 +2,6 @@
|
|
2 |
base_model: google/pegasus-cnn_dailymail
|
3 |
tags:
|
4 |
- generated_from_trainer
|
5 |
-
datasets:
|
6 |
-
- samsum
|
7 |
model-index:
|
8 |
- name: pegasus-samsum
|
9 |
results: []
|
@@ -14,9 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
14 |
|
15 |
# pegasus-samsum
|
16 |
|
17 |
-
This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the
|
18 |
-
It achieves the following results on the evaluation set:
|
19 |
-
- Loss: 1.4833
|
20 |
|
21 |
## Model description
|
22 |
|
@@ -48,14 +44,11 @@ The following hyperparameters were used during training:
|
|
48 |
|
49 |
### Training results
|
50 |
|
51 |
-
| Training Loss | Epoch | Step | Validation Loss |
|
52 |
-
|:-------------:|:------:|:----:|:---------------:|
|
53 |
-
| 1.6599 | 0.5430 | 500 | 1.4833 |
|
54 |
|
55 |
|
56 |
### Framework versions
|
57 |
|
58 |
-
- Transformers 4.42.
|
59 |
-
- Pytorch 2.3.
|
60 |
- Datasets 2.20.0
|
61 |
- Tokenizers 0.19.1
|
|
|
2 |
base_model: google/pegasus-cnn_dailymail
|
3 |
tags:
|
4 |
- generated_from_trainer
|
|
|
|
|
5 |
model-index:
|
6 |
- name: pegasus-samsum
|
7 |
results: []
|
|
|
12 |
|
13 |
# pegasus-samsum
|
14 |
|
15 |
+
This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the None dataset.
|
|
|
|
|
16 |
|
17 |
## Model description
|
18 |
|
|
|
44 |
|
45 |
### Training results
|
46 |
|
|
|
|
|
|
|
47 |
|
48 |
|
49 |
### Framework versions
|
50 |
|
51 |
+
- Transformers 4.42.4
|
52 |
+
- Pytorch 2.3.1+cu121
|
53 |
- Datasets 2.20.0
|
54 |
- Tokenizers 0.19.1
|
generation_config.json
CHANGED
@@ -9,5 +9,5 @@
|
|
9 |
"min_length": 32,
|
10 |
"num_beams": 8,
|
11 |
"pad_token_id": 0,
|
12 |
-
"transformers_version": "4.42.
|
13 |
}
|
|
|
9 |
"min_length": 32,
|
10 |
"num_beams": 8,
|
11 |
"pad_token_id": 0,
|
12 |
+
"transformers_version": "4.42.4"
|
13 |
}
|
tokenizer.json
CHANGED
@@ -1,7 +1,19 @@
|
|
1 |
{
|
2 |
"version": "1.0",
|
3 |
-
"truncation":
|
4 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
"added_tokens": [
|
6 |
{
|
7 |
"id": 0,
|
|
|
1 |
{
|
2 |
"version": "1.0",
|
3 |
+
"truncation": {
|
4 |
+
"direction": "Right",
|
5 |
+
"max_length": 1024,
|
6 |
+
"strategy": "LongestFirst",
|
7 |
+
"stride": 0
|
8 |
+
},
|
9 |
+
"padding": {
|
10 |
+
"strategy": "BatchLongest",
|
11 |
+
"direction": "Right",
|
12 |
+
"pad_to_multiple_of": null,
|
13 |
+
"pad_id": 0,
|
14 |
+
"pad_type_id": 0,
|
15 |
+
"pad_token": "<pad>"
|
16 |
+
},
|
17 |
"added_tokens": [
|
18 |
{
|
19 |
"id": 0,
|