End of training

README.md

@@ -6,6 +6,8 @@ tags:
 - sft
 - generated_from_trainer
 base_model: Viet-Mistral/Vistral-7B-Chat
+datasets:
+- generator
 model-index:
 - name: vietnamese-news-summarization-vistral-7b
   results: []
@@ -16,7 +18,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # vietnamese-news-summarization-vistral-7b
 
-This model is a fine-tuned version of [Viet-Mistral/Vistral-7B-Chat](https://huggingface.co/Viet-Mistral/Vistral-7B-Chat) on an unknown dataset.
+This model is a fine-tuned version of [Viet-Mistral/Vistral-7B-Chat](https://huggingface.co/Viet-Mistral/Vistral-7B-Chat) on the generator dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.8576
 
 ## Model description
 
@@ -44,6 +48,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_warmup_steps: 0.03
 - training_steps: 100
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss |
+|:-------------:|:------:|:----:|:---------------:|
+| 2.0431        | 0.0060 | 20   | 2.0914          |
+| 2.0513        | 0.0119 | 40   | 2.0405          |
+| 2.0366        | 0.0179 | 60   | 1.9899          |
+| 1.946         | 0.0238 | 80   | 1.9301          |
+| 1.9324        | 0.0298 | 100  | 1.8576          |
+
+
 ### Framework versions
 
 - PEFT 0.10.0