End of training

README.md

@@ -1,21 +1,43 @@
 ---
 library_name: transformers
+language:
+- hi
 license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+datasets:
+- mozilla-foundation/common_voice_11_0
+metrics:
+- wer
 model-index:
-- name: whisper-small-ft-001
-  results: []
+- name: Whisper Small Hi - Sanchit Gandhi
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: Common Voice 11.0
+      type: mozilla-foundation/common_voice_11_0
+      config: hi
+      split: None
+      args: 'config: hi, split: test'
+    metrics:
+    - name: Wer
+      type: wer
+      value: 32.341488190976044
 ---
 
 <!-- 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. -->
 
 [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/kokoorg/whisper-small-ft-001/runs/we7mrlp5)
-# whisper-small-ft-001
+# Whisper Small Hi - Sanchit Gandhi
 
-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 11.0 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.4697
+- Wer: 32.3415
 
 ## Model description
 
@@ -44,6 +66,17 @@ The following hyperparameters were used during training:
 - training_steps: 5000
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch   | Step | Validation Loss | Wer     |
+|:-------------:|:-------:|:----:|:---------------:|:-------:|
+| 0.092         | 2.4450  | 1000 | 0.2995          | 35.1562 |
+| 0.023         | 4.8900  | 2000 | 0.3540          | 33.9330 |
+| 0.0015        | 7.3350  | 3000 | 0.4215          | 32.3542 |
+| 0.0004        | 9.7800  | 4000 | 0.4573          | 32.3246 |
+| 0.0002        | 12.2249 | 5000 | 0.4697          | 32.3415 |
+
+
 ### Framework versions
 
 - Transformers 4.44.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eeb7f93bc90a026c3006432bc37cca2fb144401bc827b2bc0c729bee057b8c00
+oid sha256:f616cf2e376a325a112f564363f242e11bb72e7ca1bd1de56c22f961e2a96705
 size 966995080