End of training

README.md

@@ -16,9 +16,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 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.2178
-- Bleu Score: 78.7502
-- Gen Len: 12.7826
+- Loss: 0.2349
+- Bleu Score: 79.2125
+- Gen Len: 12.7933
 
 ## Model description
 
@@ -38,8 +38,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
-- train_batch_size: 16
-- eval_batch_size: 16
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
@@ -49,9 +49,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Bleu Score | Gen Len |
 |:-------------:|:-----:|:----:|:---------------:|:----------:|:-------:|
-| 0.4635        | 1.0   | 419  | 0.2255          | 77.2861    | 12.7969 |
-| 0.166         | 2.0   | 838  | 0.2026          | 78.3051    | 12.8041 |
-| 0.0752        | 3.0   | 1257 | 0.2178          | 78.7502    | 12.7826 |
+| 0.4686        | 1.0   | 838  | 0.2500          | 77.4621    | 12.8244 |
+| 0.1722        | 2.0   | 1676 | 0.2120          | 78.5608    | 12.7933 |
+| 0.0703        | 3.0   | 2514 | 0.2349          | 79.2125    | 12.7933 |
 
 
 ### Framework versions

logs/events.out.tfevents.1727321381.c600b8c03487.1830.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:558d80f10b849cc46fa84601c69e0d7b17480a575de306ad43a056b39d441d32
-size 6356
+oid sha256:8e0fa55b8b52d148da1ec5a227b2de8bacef2883bd45e3e85cf2a995c08214d2
+size 7297

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ecfdc3c7fca7e6166726877e3910c67b7f44393b3d5a3e3d4fc755ec4c7918f1
+oid sha256:cebdc591a3e32a392cf8efd815278c785985d8e74ea693fae84b9937758cb2be
 size 903834408