IAmSkyDra
/

BaViT5_v01

Text2Text Generation

Generated from Trainer

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

IAmSkyDra commited on 17 days ago

Commit

a17a65d

·

verified ·

1 Parent(s): 129d621

End of training

Files changed (1) hide show

README.md +19 -16

README.md CHANGED Viewed

@@ -1,7 +1,7 @@
 ---
 library_name: transformers
 license: mit
-base_model: VietAI/vit5-large
 tags:
 - generated_from_trainer
 metrics:
@@ -16,10 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 # BaViT5_v01
-This model is a fine-tuned version of [VietAI/vit5-large](https://huggingface.co/VietAI/vit5-large) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4562
-- Sacrebleu: 15.4902
 ## Model description
@@ -51,18 +51,21 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step  | Validation Loss | Sacrebleu |
 |:-------------:|:-----:|:-----:|:---------------:|:---------:|
-| 0.5323        | 1.0   | 2966  | 0.4843          | 10.8807   |
-| 0.4426        | 2.0   | 5932  | 0.4266          | 13.2481   |
-| 0.3629        | 3.0   | 8898  | 0.4084          | 14.2709   |
-| 0.3321        | 4.0   | 11864 | 0.4032          | 14.8016   |
-| 0.286         | 5.0   | 14830 | 0.4061          | 15.1102   |
-| 0.2528        | 6.0   | 17796 | 0.4160          | 15.2808   |
-| 0.2235        | 7.0   | 20762 | 0.4270          | 15.4345   |
-| 0.2018        | 8.0   | 23728 | 0.4400          | 15.4360   |
-| 0.1856        | 9.0   | 26694 | 0.4562          | 15.4902   |
-| 0.1639        | 10.0  | 29660 | 0.4705          | 15.4167   |
-| 0.1565        | 11.0  | 32626 | 0.4886          | 15.4478   |
-| 0.1392        | 12.0  | 35592 | 0.5035          | 15.4189   |
 ### Framework versions

 ---
 library_name: transformers
 license: mit
+base_model: VietAI/vit5-base
 tags:
 - generated_from_trainer
 metrics:
 # BaViT5_v01
+This model is a fine-tuned version of [VietAI/vit5-base](https://huggingface.co/VietAI/vit5-base) on the None dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.4623
+- Sacrebleu: 14.3803
 ## Model description
 | Training Loss | Epoch | Step  | Validation Loss | Sacrebleu |
 |:-------------:|:-----:|:-----:|:---------------:|:---------:|
+| 0.6515        | 1.0   | 2966  | 0.5899          | 7.7723    |
+| 0.576         | 2.0   | 5932  | 0.5257          | 10.4904   |
+| 0.4939        | 3.0   | 8898  | 0.4969          | 11.8064   |
+| 0.4842        | 4.0   | 11864 | 0.4793          | 12.5193   |
+| 0.4459        | 5.0   | 14830 | 0.4704          | 12.9876   |
+| 0.4222        | 6.0   | 17796 | 0.4632          | 13.2632   |
+| 0.4005        | 7.0   | 20762 | 0.4612          | 13.5868   |
+| 0.3869        | 8.0   | 23728 | 0.4580          | 13.8162   |
+| 0.381         | 9.0   | 26694 | 0.4556          | 13.9756   |
+| 0.3594        | 10.0  | 29660 | 0.4561          | 14.0827   |
+| 0.363         | 11.0  | 32626 | 0.4578          | 14.1701   |
+| 0.3427        | 12.0  | 35592 | 0.4591          | 14.2903   |
+| 0.3425        | 13.0  | 38558 | 0.4603          | 14.3091   |
+| 0.3377        | 14.0  | 41524 | 0.4611          | 14.3649   |
+| 0.314         | 15.0  | 44490 | 0.4623          | 14.3803   |
 ### Framework versions