update model card README.md

README.md

@@ -2,8 +2,8 @@
 license: apache-2.0
 tags:
 - generated_from_trainer
-datasets:
-- scientific_lay_summarisation
+metrics:
+- rouge
 model-index:
 - name: summarization_model
   results: []
@@ -14,19 +14,14 @@ should probably proofread and complete it, then remove this comment. -->
 
 # summarization_model
 
-This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the scientific_lay_summarisation dataset.
+This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 2.7911
-- eval_rouge1: 0.1518
-- eval_rouge2: 0.0308
-- eval_rougeL: 0.1126
-- eval_rougeLsum: 0.1127
-- eval_gen_len: 19.0
-- eval_runtime: 16.8648
-- eval_samples_per_second: 14.29
-- eval_steps_per_second: 0.949
-- epoch: 5.51
-- step: 1500
+- Loss: 0.1359
+- Rouge1: 0.1813
+- Rouge2: 0.1114
+- Rougel: 0.1616
+- Rougelsum: 0.1617
+- Gen Len: 19.0
 
 ## Model description
 
@@ -51,12 +46,26 @@ 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: 32
+- num_epochs: 8
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
+|:-------------:|:-----:|:-----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
+| 0.2358        | 1.0   | 1635  | 0.1719          | 0.1758 | 0.1033 | 0.1554 | 0.1554    | 19.0    |
+| 0.2043        | 2.0   | 3270  | 0.1574          | 0.1764 | 0.1046 | 0.1561 | 0.1561    | 19.0    |
+| 0.191         | 3.0   | 4905  | 0.1505          | 0.1778 | 0.1069 | 0.1577 | 0.1578    | 19.0    |
+| 0.178         | 4.0   | 6540  | 0.1448          | 0.1797 | 0.1093 | 0.1597 | 0.1597    | 19.0    |
+| 0.1734        | 5.0   | 8175  | 0.1406          | 0.1804 | 0.1102 | 0.1605 | 0.1604    | 19.0    |
+| 0.1681        | 6.0   | 9810  | 0.1376          | 0.1811 | 0.111  | 0.1613 | 0.1613    | 19.0    |
+| 0.1665        | 7.0   | 11445 | 0.1365          | 0.1815 | 0.1114 | 0.1618 | 0.1618    | 19.0    |
+| 0.1643        | 8.0   | 13080 | 0.1359          | 0.1813 | 0.1114 | 0.1616 | 0.1617    | 19.0    |
+
+
 ### Framework versions
 
-- Transformers 4.27.4
+- Transformers 4.28.0
 - Pytorch 2.0.0+cu118
 - Datasets 2.11.0
 - Tokenizers 0.13.3