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.2349
-- Bleu Score: 79.2125
-- Gen Len: 12.7933
+- Loss: 0.2178
+- Bleu Score: 78.7502
+- Gen Len: 12.7826
 
 ## Model description
 
@@ -38,8 +38,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
-- train_batch_size: 8
-- eval_batch_size: 8
+- train_batch_size: 16
+- eval_batch_size: 16
 - 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.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 |
+| 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 |
 
 
 ### Framework versions

logs/events.out.tfevents.1727327574.c600b8c03487.28163.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4eb7ad7c99273846f71ed37ad8f0b1550c2c20b7b78724c4f43a77af66c107a6
-size 6945
+oid sha256:9d8e6280524e0409aaadbbc8d765c88a9dc9a109766f4cffabc295c16296d468
+size 7299

model.safetensors

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