End of training

README.md

@@ -1,19 +1,38 @@
 ---
+language:
+- en
 license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+datasets:
+- jmcastelo17/FIFA_commentary
+metrics:
+- wer
 model-index:
-- name: whisper-small-hi
-  results: []
+- name: Whisper Small FIFA_commentary
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: FIFA_commentary
+      type: jmcastelo17/FIFA_commentary
+    metrics:
+    - name: Wer
+      type: wer
+      value: 24.281150159744406
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# whisper-small-hi
+# Whisper Small FIFA_commentary
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the None dataset.
+This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the FIFA_commentary dataset.
+It achieves the following results on the evaluation set:
+- Loss: 1.3163
+- Wer: 24.2812
 
 ## Model description
 
@@ -38,16 +57,24 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 10
-- training_steps: 10
+- lr_scheduler_warmup_steps: 50
+- num_epochs: 5
+- mixed_precision_training: Native AMP
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Wer     |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|
+| 5.3488        | 1.0   | 12   | 4.7962          | 46.0064 |
+| 4.6557        | 2.0   | 24   | 3.5938          | 38.9776 |
+| 3.5187        | 3.0   | 36   | 2.0149          | 30.6709 |
+| 2.2038        | 4.0   | 48   | 1.4935          | 30.0319 |
+| 1.0546        | 5.0   | 60   | 1.3163          | 24.2812 |
 
 
 ### Framework versions
 
-- Transformers 4.40.0.dev0
+- Transformers 4.39.3
 - Pytorch 2.2.1+cu121
 - Datasets 2.18.0
 - Tokenizers 0.15.2

generation_config.json

@@ -252,5 +252,5 @@
     "transcribe": 50359,
     "translate": 50358
   },
-  "transformers_version": "4.40.0.dev0"
+  "transformers_version": "4.39.3"
 }