update model card README.md

README.md

@@ -16,13 +16,13 @@ model-index:
     dataset:
       name: multi_news
       type: multi_news
-      config: train
+      config: default
       split: validation
-      args: train
+      args: default
     metrics:
     - name: Rouge1
       type: rouge
-      value: 27.57
+      value: 26.77
 ---
 
 <!-- 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.9167
-- Rouge1: 27.57
-- Rouge2: 8.53
-- Rougel: 15.17
-- Rougelsum: 18.03
+- Loss: 2.4116
+- Rouge1: 26.77
+- Rouge2: 10.28
+- Rougel: 21.26
+- Rougelsum: 22.02
 
 ## Model description
 
@@ -61,25 +61,18 @@ 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: 8
+- num_epochs: 1
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
-|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|
-| 2.8539        | 1.0   | 1250  | 2.5026          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 2.3547        | 2.0   | 2500  | 2.5102          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 2.0079        | 3.0   | 3750  | 2.5593          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 1.7303        | 4.0   | 5000  | 2.6260          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 1.4993        | 5.0   | 6250  | 2.7184          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 1.3136        | 6.0   | 7500  | 2.8246          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 1.1718        | 7.0   | 8750  | 2.8684          | 27.57  | 8.53   | 15.17  | 18.03     |
-| 1.0729        | 8.0   | 10000 | 2.9167          | 27.57  | 8.53   | 15.17  | 18.03     |
+| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|
+| 2.7976        | 1.0   | 1250 | 2.4116          | 26.77  | 10.28  | 21.26  | 22.02     |
 
 
 ### Framework versions
 
-- Transformers 4.29.2
-- Pytorch 2.0.1+cu118
+- Transformers 4.28.0
+- Pytorch 2.0.0+cu117
 - Datasets 2.12.0
 - Tokenizers 0.13.3