update model card README.md

README.md

@@ -17,12 +17,12 @@ model-index:
       name: multi_news
       type: multi_news
       config: default
-      split: train[:20%]
+      split: train[:95%]
       args: default
     metrics:
     - name: Rouge1
       type: rouge
-      value: 9.9082
+      value: 9.815
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,11 +32,11 @@ 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 the multi_news dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.5350
-- Rouge1: 9.9082
-- Rouge2: 3.6995
-- Rougel: 7.6135
-- Rougelsum: 9.0176
+- Loss: 2.3427
+- Rouge1: 9.815
+- Rouge2: 3.8774
+- Rougel: 7.6169
+- Rougelsum: 8.9863
 
 ## Model description
 
@@ -61,17 +61,15 @@ 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: 5
+- num_epochs: 3
 
 ### Training results
 
 | Training Loss | Epoch | Step  | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
 |:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
-| 2.8555        | 1.0   | 4047  | 2.5846          | 9.7797 | 3.6212 | 7.5597 | 8.9387    |
-| 2.5262        | 2.0   | 8094  | 2.5231          | 9.7969 | 3.5968 | 7.5592 | 8.9532    |
-| 2.3195        | 3.0   | 12141 | 2.5149          | 9.83   | 3.6338 | 7.5109 | 8.9725    |
-| 2.1655        | 4.0   | 16188 | 2.5188          | 9.8704 | 3.6936 | 7.6094 | 9.0336    |
-| 2.055         | 5.0   | 20235 | 2.5350          | 9.9082 | 3.6995 | 7.6135 | 9.0176    |
+| 2.75          | 1.0   | 19225 | 2.4494          | 9.5021 | 3.5429 | 7.3531 | 8.6912    |
+| 2.456         | 2.0   | 38450 | 2.3665          | 9.8103 | 3.8494 | 7.6256 | 8.9991    |
+| 2.285         | 3.0   | 57675 | 2.3427          | 9.815  | 3.8774 | 7.6169 | 8.9863    |
 
 
 ### Framework versions