update model card README.md

README.md

@@ -20,7 +20,7 @@ model-index:
     metrics:
     - name: Rouge1
       type: rouge
-      value: 0.0
+      value: 32.9341
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -30,12 +30,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [google/pegasus-large](https://huggingface.co/google/pegasus-large) on the aeslc dataset.
 It achieves the following results on the evaluation set:
-- Loss: 8.0891
-- Rouge1: 0.0
-- Rouge2: 0.0
-- Rougel: 0.0
-- Rougelsum: 0.0
-- Gen Len: 1.0
+- Loss: 3.0274
+- Rouge1: 32.9341
+- Rouge2: 17.7876
+- Rougel: 32.0858
+- Rougelsum: 32.0886
+- Gen Len: 11.6577
 
 ## Model description
 
@@ -54,9 +54,9 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0052
-- train_batch_size: 4
-- eval_batch_size: 4
+- learning_rate: 2e-05
+- train_batch_size: 2
+- eval_batch_size: 2
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -65,9 +65,9 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
-|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
-| 0.0           | 1.0   | 3609 | 8.0891          | 0.0    | 0.0    | 0.0    | 0.0       | 1.0     |
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
+| 3.3247        | 1.0   | 7218 | 3.0274          | 32.9341 | 17.7876 | 32.0858 | 32.0886   | 11.6577 |
 
 
 ### Framework versions