End of training

README.md

@@ -6,6 +6,8 @@ license: apache-2.0
 base_model: google/umt5-small
 tags:
 - generated_from_trainer
+metrics:
+- wer
 model-index:
 - name: Elite_Enigmas/Version1
   results: []
@@ -17,6 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 # Elite_Enigmas/Version1
 
 This model is a fine-tuned version of [google/umt5-small](https://huggingface.co/google/umt5-small) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 2.8451
+- Wer: 1.1562
 
 ## Model description
 
@@ -42,10 +47,14 @@ The following hyperparameters were used during training:
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 2000
-- num_epochs: 1
+- num_epochs: 10
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Wer    |
+|:-------------:|:-----:|:----:|:---------------:|:------:|
+| 7.8432        | 4.0   | 4000 | 3.8762          | 1.0    |
+| 3.0048        | 8.0   | 8000 | 2.8451          | 1.1562 |
 
 
 ### Framework versions

generation_config.json

@@ -1,5 +1,4 @@
 {
-  "_from_model_config": true,
   "decoder_start_token_id": 0,
   "eos_token_id": 1,
   "max_new_tokens": 64,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aec16986470edb56fa57800febdcaac327bc5ab04366038038cb6badc9fdcbbf
+oid sha256:7678a78c7988e9294e4bdeca59ec2c0f8d4eb16cca2417a4237906ef07a302c8
 size 1226432312

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ca4c9fe53eb85f3e629c6ba8ca7b70b49a69272e9ad28dfd4b4b168059fe2dac
+oid sha256:e14462435345526bc5d5f02e51f8db4d21dc5224cb1414829cd9956a839ffddc
 size 5432