End of training

README.md

@@ -5,9 +5,24 @@ tags:
 - generated_from_trainer
 datasets:
 - PolyAI/minds14
+metrics:
+- wer
 model-index:
 - name: whisper-tiny-ft-PolyAI-minds-14-enUS
-  results: []
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: PolyAI/minds14
+      type: PolyAI/minds14
+      config: en-US
+      split: train
+      args: en-US
+    metrics:
+    - name: Wer
+      type: wer
+      value: 0.564935064935065
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -17,14 +32,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.6973
-- eval_wer_ortho: 0.5534
-- eval_wer: 0.5537
-- eval_runtime: 30.7251
-- eval_samples_per_second: 3.678
-- eval_steps_per_second: 0.26
-- epoch: 14.29
-- step: 400
+- Loss: 0.7547
+- Wer Ortho: 0.5657
+- Wer: 0.5649
 
 ## Model description
 
@@ -48,9 +58,35 @@ The following hyperparameters were used during training:
 - eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: polynomial
+- lr_scheduler_type: constant_with_warmup
 - lr_scheduler_warmup_steps: 150
-- training_steps: 1000
+- training_steps: 500
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer    |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|
+| 3.1392        | 0.89  | 25   | 2.1394          | 0.5392    | 0.4061 |
+| 1.1631        | 1.79  | 50   | 0.6562          | 0.5848    | 0.5567 |
+| 0.4252        | 2.68  | 75   | 0.5337          | 0.6496    | 0.6352 |
+| 0.2071        | 3.57  | 100  | 0.5282          | 0.4738    | 0.4616 |
+| 0.1267        | 4.46  | 125  | 0.5674          | 0.4300    | 0.4221 |
+| 0.0503        | 5.36  | 150  | 0.5919          | 0.3924    | 0.3819 |
+| 0.0336        | 6.25  | 175  | 0.6248          | 0.3689    | 0.3548 |
+| 0.0235        | 7.14  | 200  | 0.6359          | 0.3714    | 0.3654 |
+| 0.012         | 8.04  | 225  | 0.6404          | 0.3658    | 0.3548 |
+| 0.0152        | 8.93  | 250  | 0.7101          | 0.5515    | 0.5431 |
+| 0.0081        | 9.82  | 275  | 0.6697          | 0.3763    | 0.3660 |
+| 0.0098        | 10.71 | 300  | 0.6560          | 0.3541    | 0.3465 |
+| 0.0089        | 11.61 | 325  | 0.7286          | 0.5367    | 0.5419 |
+| 0.01          | 12.5  | 350  | 0.7133          | 0.4769    | 0.4764 |
+| 0.0086        | 13.39 | 375  | 0.7350          | 0.5435    | 0.5378 |
+| 0.0062        | 14.29 | 400  | 0.7670          | 0.4275    | 0.4286 |
+| 0.0039        | 15.18 | 425  | 0.7543          | 0.4553    | 0.4534 |
+| 0.0086        | 16.07 | 450  | 0.7541          | 0.4571    | 0.4551 |
+| 0.0043        | 16.96 | 475  | 0.7642          | 0.5527    | 0.5525 |
+| 0.0022        | 17.86 | 500  | 0.7547          | 0.5657    | 0.5649 |
+
 
 ### Framework versions
 

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:270f5b6c5658adabe1ecceec76337b4a7e00b639f0557ea2386a0e179db0b14f
+oid sha256:cebf112b1eff3ebc1121e2ffcf99394f09d8a35f0e4c79b58718828ff520b2a5
 size 151097459

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a76cd3ccb55af5c02a02de7350257ffb38591eae73f9c0fa1ca52eca6773cc38
+oid sha256:cef9eff0f26cfd56f369720eef2f6ea99d2284cf658ec29b2035ac484081295c
 size 4143