Training complete

README.md

@@ -1,6 +1,6 @@
 ---
 license: apache-2.0
-base_model: t5-small
+base_model: t5-base
 tags:
 - text2text-generation
 - generated_from_trainer
@@ -16,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vi-correction-t5
 
-This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on an unknown dataset.
+This model is a fine-tuned version of [t5-base](https://huggingface.co/t5-base) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2966
-- Sacrebleu: 0.0
+- Loss: 0.1465
+- Sacrebleu: 0.4788
 
 ## Model description
 
@@ -52,10 +52,10 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Sacrebleu |
 |:-------------:|:------:|:----:|:---------------:|:---------:|
-| No log        | 0.992  | 93   | 0.4620          | 0.0       |
-| No log        | 1.9947 | 187  | 0.3466          | 0.0       |
-| No log        | 2.9973 | 281  | 0.3070          | 0.0       |
-| No log        | 3.968  | 372  | 0.2966          | 0.0       |
+| No log        | 0.992  | 93   | 0.1898          | 0.1884    |
+| No log        | 1.9947 | 187  | 0.1586          | 0.4429    |
+| No log        | 2.9973 | 281  | 0.1491          | 0.4715    |
+| No log        | 3.968  | 372  | 0.1465          | 0.4788    |
 
 
 ### Framework versions