nandavikas16
commited on
Commit
•
51173a9
1
Parent(s):
39a8e04
Model save
Browse files
README.md
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
---
|
|
|
2 |
license: mit
|
3 |
base_model: facebook/bart-large-cnn
|
4 |
tags:
|
@@ -17,11 +18,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 +53,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.
|
71 |
-
- Datasets
|
72 |
- Tokenizers 0.19.1
|
|
|
1 |
---
|
2 |
+
library_name: transformers
|
3 |
license: mit
|
4 |
base_model: facebook/bart-large-cnn
|
5 |
tags:
|
|
|
18 |
|
19 |
This model is a fine-tuned version of [facebook/bart-large-cnn](https://huggingface.co/facebook/bart-large-cnn) on an unknown dataset.
|
20 |
It achieves the following results on the evaluation set:
|
21 |
+
- Loss: 0.1120
|
22 |
+
- Rouge1: 51.232
|
23 |
+
- Rouge2: 37.3103
|
24 |
+
- Rougel: 39.2783
|
25 |
+
- Rougelsum: 39.2011
|
26 |
|
27 |
## Model description
|
28 |
|
|
|
53 |
|
54 |
| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
|
55 |
|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
|
56 |
+
| 0.6379 | 1.0 | 40 | 0.2289 | 45.9991 | 29.5151 | 34.3864 | 34.3984 |
|
57 |
+
| 0.2731 | 2.0 | 80 | 0.1935 | 47.3991 | 33.1933 | 38.1538 | 38.0514 |
|
58 |
+
| 0.2362 | 3.0 | 120 | 0.1734 | 47.4125 | 32.2496 | 35.7852 | 35.8279 |
|
59 |
+
| 0.222 | 4.0 | 160 | 0.1665 | 46.2226 | 32.0249 | 37.016 | 36.8941 |
|
60 |
+
| 0.2005 | 5.0 | 200 | 0.1530 | 50.1647 | 35.1015 | 39.0526 | 39.0721 |
|
61 |
+
| 0.1971 | 6.0 | 240 | 0.1434 | 49.7914 | 35.5371 | 39.2372 | 39.244 |
|
62 |
+
| 0.1754 | 7.0 | 280 | 0.1286 | 49.8482 | 35.7536 | 40.2412 | 40.2248 |
|
63 |
+
| 0.1777 | 8.0 | 320 | 0.1187 | 51.6342 | 38.223 | 41.4109 | 41.3626 |
|
64 |
+
| 0.1555 | 9.0 | 360 | 0.1149 | 49.1858 | 36.1404 | 38.857 | 38.7268 |
|
65 |
+
| 0.1415 | 10.0 | 400 | 0.1120 | 51.232 | 37.3103 | 39.2783 | 39.2011 |
|
66 |
|
67 |
|
68 |
### Framework versions
|
69 |
|
70 |
+
- Transformers 4.44.2
|
71 |
+
- Pytorch 2.2.0+cu121
|
72 |
+
- Datasets 3.0.0
|
73 |
- 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.44.2",
|
15 |
"use_cache": false
|
16 |
}
|
runs/Sep11_13-54-28_orion/events.out.tfevents.1726062921.orion.2195743.0
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7ea152f06461caaaf648db094fd08906b2c356d0e53a9da15bd8b066001ff15d
|
3 |
+
size 13333
|
runs/Sep11_13-54-28_orion/events.out.tfevents.1726064648.orion.2195743.1
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7b5e72ae02ea59c8d57b911498b3f7867748a3c000e9c87e3084f0307fd409e2
|
3 |
+
size 562
|