update model card README.md

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: google/flan-t5-base
+base_model: google/flan-t5-small
 tags:
 - generated_from_trainer
 metrics:
@@ -15,13 +15,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # subset_model_flan_t5
 
-This model is a fine-tuned version of [google/flan-t5-base](https://huggingface.co/google/flan-t5-base) on the None dataset.
+This model is a fine-tuned version of [google/flan-t5-small](https://huggingface.co/google/flan-t5-small) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 1.2027
+- Loss: 1.2878
 - Rouge1: 0.2857
 - Rouge2: 0.15
-- Rougel: 0.2381
-- Rougelsum: 0.2381
+- Rougel: 0.2857
+- Rougelsum: 0.2857
 - Gen Len: 19.0
 
 ## Model description
@@ -53,10 +53,10 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:---------:|:-------:|
-| No log        | 1.0   | 1    | 1.4603          | 0.2857 | 0.15   | 0.2381 | 0.2381    | 19.0    |
-| No log        | 2.0   | 2    | 1.2954          | 0.2857 | 0.15   | 0.2381 | 0.2381    | 19.0    |
-| No log        | 3.0   | 3    | 1.2214          | 0.2857 | 0.15   | 0.2381 | 0.2381    | 19.0    |
-| No log        | 4.0   | 4    | 1.2027          | 0.2857 | 0.15   | 0.2381 | 0.2381    | 19.0    |
+| No log        | 1.0   | 1    | 1.3451          | 0.2857 | 0.15   | 0.2857 | 0.2857    | 19.0    |
+| No log        | 2.0   | 2    | 1.3067          | 0.2857 | 0.15   | 0.2857 | 0.2857    | 19.0    |
+| No log        | 3.0   | 3    | 1.2925          | 0.2857 | 0.15   | 0.2857 | 0.2857    | 19.0    |
+| No log        | 4.0   | 4    | 1.2878          | 0.2857 | 0.15   | 0.2857 | 0.2857    | 19.0    |
 
 
 ### Framework versions