update model card README.md

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: facebook/bart-base
 tags:
 - generated_from_trainer
+metrics:
+- rouge
 model-index:
 - name: bart-base-finetuned-xsum
   results: []
@@ -15,17 +17,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/bart-base](https://huggingface.co/facebook/bart-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 1.7777
-- eval_rouge1: 10.4738
-- eval_rouge2: 5.5988
-- eval_rougeL: 8.9936
-- eval_rougeLsum: 10.0252
-- eval_gen_len: 20.0
-- eval_runtime: 223.3342
-- eval_samples_per_second: 4.487
-- eval_steps_per_second: 2.243
-- epoch: 5.0
-- step: 10015
+- Loss: 1.7805
+- Rouge1: 10.3678
+- Rouge2: 5.6385
+- Rougel: 9.0139
+- Rougelsum: 9.9686
+- Gen Len: 20.0
 
 ## Model description
 
@@ -50,7 +47,14 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 10
+- num_epochs: 1
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2 | Rougel | Rougelsum | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|:------:|:---------:|:-------:|
+| 1.455         | 1.0   | 2003 | 1.7805          | 10.3678 | 5.6385 | 9.0139 | 9.9686    | 20.0    |
+
 
 ### Framework versions