Training complete

README.md

@@ -5,6 +5,8 @@ base_model: facebook/mbart-large-50
 tags:
 - simplification
 - generated_from_trainer
+metrics:
+- bleu
 model-index:
 - name: mbart-neutralization
   results: []
@@ -17,14 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [facebook/mbart-large-50](https://huggingface.co/facebook/mbart-large-50) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.0200
-- eval_bleu: 95.8347
-- eval_gen_len: 18.8125
-- eval_runtime: 12.3487
-- eval_samples_per_second: 7.774
-- eval_steps_per_second: 0.972
-- epoch: 1.0
-- step: 440
+- Loss: 0.0123
+- Bleu: 97.5887
+- Gen Len: 18.75
 
 ## Model description
 
@@ -51,9 +48,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 2
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
+| No log        | 1.0   | 440  | 0.0231          | 89.8494 | 18.4792 |
+| 0.2249        | 2.0   | 880  | 0.0123          | 97.5887 | 18.75   |
+
+
 ### Framework versions
 
 - Transformers 4.49.0
-- Pytorch 2.5.1+cu124
-- Datasets 3.3.2
-- Tokenizers 0.21.0
+- Pytorch 2.6.0+cu124
+- Datasets 3.4.1
+- Tokenizers 0.21.1