End of training

README.md

@@ -1,19 +1,41 @@
 ---
+language:
+- yue
 license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+datasets:
+- mozilla-foundation/common_voice_17_0
+metrics:
+- wer
 model-index:
-- name: whisper-small-cantonese_02-05-2024-1727
-  results: []
+- name: Whisper Small Cantanese
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: Common Voice 17.0
+      type: mozilla-foundation/common_voice_17_0
+      config: yue
+      split: None
+      args: 'config: yue, split: test'
+    metrics:
+    - name: Wer
+      type: wer
+      value: 55.57641757416447
 ---
 
 <!-- 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-cantonese_02-05-2024-1727
+# Whisper Small Cantanese
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on an unknown dataset.
+This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the Common Voice 17.0 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2823
+- Wer: 55.5764
 
 ## Model description
 
@@ -44,6 +66,16 @@ The following hyperparameters were used during training:
 - training_steps: 4000
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Wer     |
+|:-------------:|:------:|:----:|:---------------:|:-------:|
+| 0.116         | 1.3908 | 1000 | 0.2522          | 59.0687 |
+| 0.0418        | 2.7816 | 2000 | 0.2504          | 57.7169 |
+| 0.0065        | 4.1725 | 3000 | 0.2702          | 56.1772 |
+| 0.0031        | 5.5633 | 4000 | 0.2823          | 55.5764 |
+
+
 ### Framework versions
 
 - Transformers 4.40.1