update model card README.md

README.md

@@ -17,11 +17,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 2.3633
-- Rouge1: 40.5438
-- Rouge2: 17.4543
-- Rougel: 35.3889
-- Rougelsum: 35.8547
+- Loss: 3.9358
+- Rouge1: 30.5741
+- Rouge2: 11.5241
+- Rougel: 25.7323
+- Rougelsum: 26.7256
 
 ## Model description
 
@@ -40,23 +40,28 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 5.6e-05
-- train_batch_size: 10
-- eval_batch_size: 10
+- learning_rate: 1e-06
+- train_batch_size: 20
+- eval_batch_size: 20
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 5
+- num_epochs: 10
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
-| 3.2731        | 1.0   | 36   | 2.7149          | 31.9322 | 12.2221 | 27.5812 | 27.7701   |
-| 2.7097        | 2.0   | 72   | 2.4906          | 38.1302 | 15.5635 | 32.8281 | 33.1249   |
-| 2.5515        | 3.0   | 108  | 2.4084          | 40.3328 | 17.3795 | 34.7863 | 35.0262   |
-| 2.4601        | 4.0   | 144  | 2.3730          | 40.4377 | 17.2083 | 35.3265 | 35.7168   |
-| 2.377         | 5.0   | 180  | 2.3633          | 40.5438 | 17.4543 | 35.3889 | 35.8547   |
+| 3.9531        | 1.0   | 18   | 4.0768          | 30.8668 | 12.0448 | 26.0748 | 27.0649   |
+| 3.8999        | 2.0   | 36   | 4.0458          | 30.9565 | 11.9145 | 26.1991 | 27.1728   |
+| 3.8301        | 3.0   | 54   | 4.0187          | 30.9917 | 11.8826 | 26.1494 | 27.1447   |
+| 3.8494        | 4.0   | 72   | 3.9960          | 30.6865 | 11.5462 | 25.9455 | 26.8867   |
+| 3.9007        | 5.0   | 90   | 3.9776          | 30.6963 | 11.549  | 25.9559 | 26.8594   |
+| 3.7678        | 6.0   | 108  | 3.9624          | 30.6872 | 11.599  | 25.9522 | 26.8569   |
+| 3.8246        | 7.0   | 126  | 3.9509          | 30.7149 | 11.6052 | 25.9879 | 26.861    |
+| 3.8238        | 8.0   | 144  | 3.9425          | 30.5069 | 11.3856 | 25.6649 | 26.6549   |
+| 3.7948        | 9.0   | 162  | 3.9374          | 30.5871 | 11.5312 | 25.743  | 26.6969   |
+| 3.807         | 10.0  | 180  | 3.9358          | 30.5741 | 11.5241 | 25.7323 | 26.7256   |
 
 
 ### Framework versions