update model card README.md

README.md

@@ -10,7 +10,7 @@ datasets:
 metrics:
 - wer
 model-index:
-- name: Whisper tiny ID - Augmented
+- name: Whisper base ID - Augmented
   results:
   - task:
       name: Automatic Speech Recognition
@@ -24,18 +24,18 @@ model-index:
     metrics:
     - name: Wer
       type: wer
-      value: 14.42705294401177
+      value: 11.20271950074837
 ---
 
 <!-- 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 tiny ID - Augmented
+# Whisper base ID - Augmented
 
-This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the mozilla-foundation/common_voice_12_0 dataset.
+This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the mozilla-foundation/common_voice_12_0 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.2874
-- Wer: 14.4271
+- Loss: 0.2277
+- Wer: 11.2027
 
 ## Model description
 
@@ -70,11 +70,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Wer     |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|
-| 0.6729        | 0.77  | 200  | 0.4170          | 21.8473 |
-| 0.5012        | 1.54  | 400  | 0.3465          | 18.2560 |
-| 0.3832        | 2.31  | 600  | 0.3097          | 16.0971 |
-| 0.2853        | 3.07  | 800  | 0.2970          | 14.3180 |
-| 0.2416        | 3.84  | 1000 | 0.2874          | 14.4271 |
+| 0.5509        | 0.77  | 200  | 0.3243          | 13.8588 |
+| 0.3844        | 1.54  | 400  | 0.2590          | 11.1782 |
+| 0.259         | 2.31  | 600  | 0.2422          | 18.6813 |
+| 0.1608        | 3.07  | 800  | 0.2361          | 11.6678 |
+| 0.1205        | 3.84  | 1000 | 0.2277          | 11.2027 |
 
 
 ### Framework versions

runs/Apr17_15-27-04_dsai-server/events.out.tfevents.1681720029.dsai-server

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2cc7823e113b7ad81ce40214e3bb47e042dff0b28b575ff482cd88f8582f3cbc
-size 7392
+oid sha256:226c962edb5c007efef22acb9afb1965cf6f825225ff6296c7272a8c93b28868
+size 7549