Cathaysa
/

mbart-neutralization

text2text-generation

Generated from Trainer

Model card Files Files and versions

Metrics Training metrics Community

Cathaysa commited on Feb 17, 2024

Commit

b9d7f81

·

verified ·

1 Parent(s): d7a36c5

Training complete

Files changed (2) hide show

README.md +5 -5
model.safetensors +1 -1

README.md CHANGED Viewed

@@ -18,9 +18,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:
-- Loss: 1.8061
-- Bleu: 63.7292
-- Gen Len: 26.9254
 ## Model description
@@ -51,8 +51,8 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
-| No log        | 1.0   | 34   | 2.0386          | 64.2529 | 27.4478 |
-| No log        | 2.0   | 68   | 1.8061          | 63.7292 | 26.9254 |
 ### Framework versions

 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:
+- Loss: 0.0181
+- Bleu: 98.7341
+- Gen Len: 18.4896
 ## Model description
 | Training Loss | Epoch | Step | Validation Loss | Bleu    | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
+| No log        | 1.0   | 440  | 0.0307          | 91.2911 | 18.25   |
+| 0.2343        | 2.0   | 880  | 0.0181          | 98.7341 | 18.4896 |
 ### Framework versions

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:82b99059e8a57dff52042a00e25d4bad26516a7860a6c6a7eec46d08d1ec4894
 size 2444578688

 version https://git-lfs.github.com/spec/v1
+oid sha256:be81fdab8cada8f8cdd8d57dfd898b9eeb36743291ee9865ff988dadb47b2efc
 size 2444578688