update model card README.md

README.md

@@ -1,6 +1,8 @@
 ---
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: bert2bert-cnn_dailymail-fp16-finetuned-1.0.0
   results: []
@@ -13,16 +15,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 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.
 It achieves the following results on the evaluation set:
-- eval_loss: 2.4478
-- eval_rouge1: 45.3108
-- eval_rouge2: 18.3254
-- eval_rougeL: 29.71
-- eval_rougeLsum: 41.2888
-- eval_runtime: 15495.5669
-- eval_samples_per_second: 0.038
-- eval_steps_per_second: 0.009
-- epoch: 1.0
-- step: 586
+- Loss: 2.3346
+- Rouge1: 46.3609
+- Rouge2: 18.8105
+- Rougel: 30.215
+- Rougelsum: 42.3642
 
 ## Model description
 
@@ -49,6 +46,14 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 2
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:------:|:---------:|
+| 2.8263        | 1.0   | 586  | 2.4478          | 45.3367 | 18.3604 | 29.713 | 41.2805   |
+| 2.1264        | 2.0   | 1172 | 2.3346          | 46.3609 | 18.8105 | 30.215 | 42.3642   |
+
+
 ### Framework versions
 
 - Transformers 4.26.1