update model card README.md

README.md

@@ -3,17 +3,20 @@ license: apache-2.0
 tags:
 - generated_from_trainer
 datasets:
-- billsum
+- cnn_dailymail
 model-index:
 - name: t5-small-medicalnews-summarization
   results: []
 ---
 
-This model is not completed yet
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
 
 # t5-small-medicalnews-summarization
 
-This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the billsum dataset.
+This model is a fine-tuned version of [weijiahaha/t5-small-medicalnews-summarization](https://huggingface.co/weijiahaha/t5-small-medicalnews-summarization) on the cnn_dailymail dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.6477
 
 ## Model description
 
@@ -32,7 +35,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
+- learning_rate: 0.0002
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
@@ -45,12 +48,12 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss |
 |:-------------:|:-----:|:----:|:---------------:|
-| No log        | 1.0   | 62   | 3.1698          |
+| 1.9195        | 1.0   | 718  | 1.6477          |
 
 
 ### Framework versions
 
-- Transformers 4.20.1
+- Transformers 4.21.1
 - Pytorch 1.12.0+cu113
 - Datasets 2.4.0
 - Tokenizers 0.12.1