End of training

README.md

@@ -1,22 +1,43 @@
 ---
 library_name: transformers
+language:
+- dv
 license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
 datasets:
-- common_voice_13_0
+- mozilla-foundation/common_voice_13_0
+metrics:
+- wer
 model-index:
-- name: whisper-small-dv
-  results: []
+- name: Whisper Small Dv - Tanishq Srivastava
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: Common Voice 13
+      type: mozilla-foundation/common_voice_13_0
+      config: dv
+      split: test
+      args: dv
+    metrics:
+    - name: Wer
+      type: wer
+      value: 18.070382863302846
 ---
 
 <!-- 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-dv
+# Whisper Small Dv - Tanishq Srivastava
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the common_voice_13_0 dataset.
+This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the Common Voice 13 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2420
+- Wer Ortho: 73.3965
+- Wer: 18.0704
 
 ## Model description
 
@@ -36,7 +57,7 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 16
+- train_batch_size: 4
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
@@ -45,9 +66,16 @@ The following hyperparameters were used during training:
 - training_steps: 500
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch  | Step | Validation Loss | Wer Ortho | Wer     |
+|:-------------:|:------:|:----:|:---------------:|:---------:|:-------:|
+| 0.1763        | 0.4078 | 500  | 0.2420          | 73.3965   | 18.0704 |
+
+
 ### Framework versions
 
-- Transformers 4.53.0
+- Transformers 4.53.1
 - Pytorch 2.6.0+cu124
 - Datasets 3.6.0
 - Tokenizers 0.21.2

generation_config.json

@@ -258,5 +258,5 @@
     "transcribe": 50359,
     "translate": 50358
   },
-  "transformers_version": "4.53.0"
+  "transformers_version": "4.53.1"
 }