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.0567
21
- - Rouge1: 55.8275
22
- - Rouge2: 45.1893
23
- - Rougel: 50.0441
24
- - Rougelsum: 50.1406
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.4872 | 1.0 | 31 | 0.1259 | 40.3668 | 25.4156 | 32.4952 | 32.5066 |
56
- | 0.1349 | 2.0 | 62 | 0.1071 | 38.4137 | 23.9982 | 29.3169 | 29.3606 |
57
- | 0.128 | 3.0 | 93 | 0.1000 | 45.61 | 29.904 | 36.2931 | 36.1606 |
58
- | 0.1138 | 4.0 | 124 | 0.0993 | 47.4837 | 33.2177 | 40.1978 | 40.0854 |
59
- | 0.1064 | 5.0 | 155 | 0.0832 | 48.4985 | 34.9247 | 42.5024 | 42.5047 |
60
- | 0.0978 | 6.0 | 186 | 0.0792 | 50.4492 | 37.1342 | 42.4463 | 42.2735 |
61
- | 0.0941 | 7.0 | 217 | 0.0726 | 53.0967 | 41.2303 | 45.9412 | 45.9511 |
62
- | 0.0748 | 8.0 | 248 | 0.0653 | 54.9193 | 43.316 | 46.8554 | 46.7847 |
63
- | 0.0791 | 9.0 | 279 | 0.0582 | 54.424 | 44.2063 | 47.1231 | 47.0471 |
64
- | 0.0742 | 10.0 | 310 | 0.0567 | 55.8275 | 45.1893 | 50.0441 | 50.1406 |
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
 
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.43.2",
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:11453e997402ffbcf0b0f3c0c3586b9d103267ad9279b9abb0a69dd94a49f98d
3
- size 12505
 
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