0xhaz commited on
Commit
be25234
1 Parent(s): 73c0a89

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -10
README.md CHANGED
@@ -1,6 +1,8 @@
1
  ---
2
  tags:
3
  - generated_from_trainer
 
 
4
  model-index:
5
  - name: bert2bert-cnn_dailymail-fp16-finetuned-1.0.0
6
  results: []
@@ -13,16 +15,11 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  This model is a fine-tuned version of [patrickvonplaten/bert2bert-cnn_dailymail-fp16](https://huggingface.co/patrickvonplaten/bert2bert-cnn_dailymail-fp16) on an unknown dataset.
15
  It achieves the following results on the evaluation set:
16
- - eval_loss: 2.4478
17
- - eval_rouge1: 45.3108
18
- - eval_rouge2: 18.3254
19
- - eval_rougeL: 29.71
20
- - eval_rougeLsum: 41.2888
21
- - eval_runtime: 15495.5669
22
- - eval_samples_per_second: 0.038
23
- - eval_steps_per_second: 0.009
24
- - epoch: 1.0
25
- - step: 586
26
 
27
  ## Model description
28
 
@@ -49,6 +46,14 @@ The following hyperparameters were used during training:
49
  - lr_scheduler_type: linear
50
  - num_epochs: 2
51
 
 
 
 
 
 
 
 
 
52
  ### Framework versions
53
 
54
  - Transformers 4.26.1
 
1
  ---
2
  tags:
3
  - generated_from_trainer
4
+ metrics:
5
+ - rouge
6
  model-index:
7
  - name: bert2bert-cnn_dailymail-fp16-finetuned-1.0.0
8
  results: []
 
15
 
16
  This model is a fine-tuned version of [patrickvonplaten/bert2bert-cnn_dailymail-fp16](https://huggingface.co/patrickvonplaten/bert2bert-cnn_dailymail-fp16) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 2.3346
19
+ - Rouge1: 46.3609
20
+ - Rouge2: 18.8105
21
+ - Rougel: 30.215
22
+ - Rougelsum: 42.3642
 
 
 
 
 
23
 
24
  ## Model description
25
 
 
46
  - lr_scheduler_type: linear
47
  - num_epochs: 2
48
 
49
+ ### Training results
50
+
51
+ | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
52
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:------:|:---------:|
53
+ | 2.8263 | 1.0 | 586 | 2.4478 | 45.3367 | 18.3604 | 29.713 | 41.2805 |
54
+ | 2.1264 | 2.0 | 1172 | 2.3346 | 46.3609 | 18.8105 | 30.215 | 42.3642 |
55
+
56
+
57
  ### Framework versions
58
 
59
  - Transformers 4.26.1