lvwerra HF staff commited on
Commit
72dd5ca
1 Parent(s): 23a73f1

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -14,6 +14,8 @@ should probably proofread and complete it, then remove this comment. -->
14
  # pegasus-samsum
15
 
16
  This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the samsum dataset.
 
 
17
 
18
  ## Model description
19
 
@@ -39,10 +41,13 @@ The following hyperparameters were used during training:
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
40
  - lr_scheduler_type: linear
41
  - lr_scheduler_warmup_steps: 500
42
- - num_epochs: 0.01
43
 
44
  ### Training results
45
 
 
 
 
46
 
47
 
48
  ### Framework versions
 
14
  # pegasus-samsum
15
 
16
  This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the samsum dataset.
17
+ It achieves the following results on the evaluation set:
18
+ - Loss: 1.6488
19
 
20
  ## Model description
21
 
 
41
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
42
  - lr_scheduler_type: linear
43
  - lr_scheduler_warmup_steps: 500
44
+ - num_epochs: 0.05
45
 
46
  ### Training results
47
 
48
+ | Training Loss | Epoch | Step | Validation Loss |
49
+ |:-------------:|:-----:|:----:|:---------------:|
50
+ | 1.6092 | 0.03 | 500 | 1.6488 |
51
 
52
 
53
  ### Framework versions