Gabriel commited on
Commit
83ab60e
1 Parent(s): 6a96ced

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -16,12 +16,12 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  This model is a fine-tuned version of [Gabriel/bart-base-cnn-swe](https://huggingface.co/Gabriel/bart-base-cnn-swe) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
- - Loss: 2.0253
20
- - Rouge1: 22.0568
21
- - Rouge2: 10.3302
22
- - Rougel: 18.0648
23
- - Rougelsum: 20.7482
24
- - Gen Len: 19.9996
25
 
26
  ## Model description
27
 
@@ -40,7 +40,7 @@ More information needed
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
- - learning_rate: 2e-05
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 42
@@ -55,13 +55,13 @@ The following hyperparameters were used during training:
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
57
  |:-------------:|:-----:|:-----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
58
- | 2.2349 | 1.0 | 17944 | 2.0643 | 21.9564 | 10.2133 | 17.9958 | 20.6502 | 19.9992 |
59
- | 2.0726 | 2.0 | 35888 | 2.0253 | 22.0568 | 10.3302 | 18.0648 | 20.7482 | 19.9996 |
60
 
61
 
62
  ### Framework versions
63
 
64
- - Transformers 4.22.0
65
  - Pytorch 1.12.1+cu113
66
  - Datasets 2.4.0
67
  - Tokenizers 0.12.1
 
16
 
17
  This model is a fine-tuned version of [Gabriel/bart-base-cnn-swe](https://huggingface.co/Gabriel/bart-base-cnn-swe) on the None dataset.
18
  It achieves the following results on the evaluation set:
19
+ - Loss: 1.9598
20
+ - Rouge1: 22.1847
21
+ - Rouge2: 10.4718
22
+ - Rougel: 18.2296
23
+ - Rougelsum: 20.8603
24
+ - Gen Len: 19.9964
25
 
26
  ## Model description
27
 
 
40
  ### Training hyperparameters
41
 
42
  The following hyperparameters were used during training:
43
+ - learning_rate: 5e-05
44
  - train_batch_size: 8
45
  - eval_batch_size: 8
46
  - seed: 42
 
55
 
56
  | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
57
  |:-------------:|:-----:|:-----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
58
+ | 2.0969 | 1.0 | 17944 | 2.0137 | 22.0653 | 10.2668 | 18.0992 | 20.7207 | 19.9984 |
59
+ | 1.904 | 2.0 | 35888 | 1.9598 | 22.1847 | 10.4718 | 18.2296 | 20.8603 | 19.9964 |
60
 
61
 
62
  ### Framework versions
63
 
64
+ - Transformers 4.22.1
65
  - Pytorch 1.12.1+cu113
66
  - Datasets 2.4.0
67
  - Tokenizers 0.12.1