update model card README.md

README.md

@@ -16,11 +16,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5912
-- Rouge1: 56.6081
-- Rouge2: 31.7858
-- Rougel: 48.4959
-- Rougelsum: 56.6806
+- Loss: 0.0225
+- Rouge1: 98.9126
+- Rouge2: 96.9479
+- Rougel: 97.9209
+- Rougelsum: 98.9061
 
 ## Model description
 
@@ -40,21 +40,28 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 32
-- eval_batch_size: 32
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 10
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|
-| 5.3993        | 1.0   | 188  | 1.4868          | 23.2843 | 6.9787  | 22.46   | 23.2664   |
-| 1.2384        | 2.0   | 376  | 0.7450          | 47.5051 | 23.5773 | 42.2383 | 47.5388   |
-| 0.8417        | 3.0   | 564  | 0.5912          | 56.6081 | 31.7858 | 48.4959 | 56.6806   |
+| 3.2555        | 1.0   | 375  | 0.7613          | 41.6446 | 20.4343 | 38.0279 | 41.5954   |
+| 0.679         | 2.0   | 750  | 0.3463          | 72.8071 | 48.9808 | 60.7026 | 72.8052   |
+| 0.4088        | 3.0   | 1125 | 0.1948          | 85.3976 | 65.3267 | 74.3572 | 85.3705   |
+| 0.2795        | 4.0   | 1500 | 0.1098          | 91.8037 | 78.9948 | 85.9716 | 91.8695   |
+| 0.204         | 5.0   | 1875 | 0.0776          | 94.6475 | 84.3954 | 89.9382 | 94.6349   |
+| 0.1544        | 6.0   | 2250 | 0.0454          | 97.197  | 91.932  | 94.8966 | 97.1919   |
+| 0.1212        | 7.0   | 2625 | 0.0384          | 97.5777 | 93.2443 | 95.4839 | 97.5692   |
+| 0.1037        | 8.0   | 3000 | 0.0315          | 97.8918 | 95.2195 | 96.8449 | 97.9063   |
+| 0.0942        | 9.0   | 3375 | 0.0253          | 98.6234 | 96.5271 | 97.6489 | 98.6284   |
+| 0.0823        | 10.0  | 3750 | 0.0225          | 98.9126 | 96.9479 | 97.9209 | 98.9061   |
 
 
 ### Framework versions