smrynrz20 commited on
Commit
90b3179
1 Parent(s): 5ebd224

End of training

Browse files
README.md CHANGED
@@ -17,12 +17,12 @@ should probably proofread and complete it, then remove this comment. -->
17
 
18
  This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
- - Loss: 1.5285
21
- - Rouge1: 0.4707
22
- - Rouge2: 0.2409
23
- - Rougel: 0.4
24
- - Rougelsum: 0.3996
25
- - Gen Len: 18.16
26
 
27
  ## Model description
28
 
@@ -41,7 +41,7 @@ More information needed
41
  ### Training hyperparameters
42
 
43
  The following hyperparameters were used during training:
44
- - learning_rate: 3.3770685983138454e-05
45
  - train_batch_size: 8
46
  - eval_batch_size: 8
47
  - seed: 42
@@ -54,14 +54,14 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
56
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
57
- | 1.8264 | 1.0 | 1842 | 1.6005 | 0.4553 | 0.2248 | 0.3854 | 0.385 | 18.4786 |
58
- | 1.5427 | 2.0 | 3684 | 1.5590 | 0.4578 | 0.227 | 0.3887 | 0.3885 | 18.2747 |
59
- | 1.3733 | 3.0 | 5526 | 1.5285 | 0.4707 | 0.2409 | 0.4 | 0.3996 | 18.16 |
60
 
61
 
62
  ### Framework versions
63
 
64
- - Transformers 4.36.2
65
  - Pytorch 2.1.0+cu121
66
- - Datasets 2.16.1
67
- - Tokenizers 0.15.0
 
17
 
18
  This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on an unknown dataset.
19
  It achieves the following results on the evaluation set:
20
+ - Loss: 1.5246
21
+ - Rouge1: 0.472
22
+ - Rouge2: 0.2373
23
+ - Rougel: 0.3986
24
+ - Rougelsum: 0.3987
25
+ - Gen Len: 18.1954
26
 
27
  ## Model description
28
 
 
41
  ### Training hyperparameters
42
 
43
  The following hyperparameters were used during training:
44
+ - learning_rate: 2.8410039143956672e-05
45
  - train_batch_size: 8
46
  - eval_batch_size: 8
47
  - seed: 42
 
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
56
  |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
57
+ | 1.822 | 1.0 | 1842 | 1.5989 | 0.4581 | 0.2227 | 0.3846 | 0.385 | 18.1893 |
58
+ | 1.5641 | 2.0 | 3684 | 1.5492 | 0.4661 | 0.2328 | 0.3936 | 0.3936 | 18.2576 |
59
+ | 1.4168 | 3.0 | 5526 | 1.5246 | 0.472 | 0.2373 | 0.3986 | 0.3987 | 18.1954 |
60
 
61
 
62
  ### Framework versions
63
 
64
+ - Transformers 4.37.2
65
  - Pytorch 2.1.0+cu121
66
+ - Datasets 2.17.0
67
+ - Tokenizers 0.15.1
generation_config.json CHANGED
@@ -8,5 +8,5 @@
8
  "no_repeat_ngram_size": 3,
9
  "num_beams": 4,
10
  "pad_token_id": 1,
11
- "transformers_version": "4.36.2"
12
  }
 
8
  "no_repeat_ngram_size": 3,
9
  "num_beams": 4,
10
  "pad_token_id": 1,
11
+ "transformers_version": "4.37.2"
12
  }
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:09d16c478850c29631f0efb639946ed216d9ad898ccca9d60ded94b133e6aa8c
3
  size 557912620
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19155afb2451a2adfe2448a0b48bac40203f0ee50f24a8208dda6fb75646d3b7
3
  size 557912620
runs/Feb12_11-22-27_4d49022648cd/events.out.tfevents.1707736948.4d49022648cd.3001.10 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d41dc9f93f0586867ae20e25c9d8448bcc2a777060a9fac7e210a50a843ab6d0
3
- size 8390
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca52f0dee8ceac25ebfd875480dbd24e02a4f8ad477647602c7cda9cb488ac38
3
+ size 9269