End of training

README.md

@@ -9,19 +9,19 @@ tags:
 metrics:
 - wer
 model-index:
-- name: openai/whisper-medium
+- name: Whisper Tiny
   results: []
 ---
 
 <!-- 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. -->
 
-# openai/whisper-medium
+# Whisper Tiny
 
 This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the Personal - Mimic Recording dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1558
-- Wer: 0.0674
+- Loss: 0.1404
+- Wer: 0.0645
 
 ## Model description
 
@@ -40,15 +40,15 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
-- train_batch_size: 4
-- eval_batch_size: 4
+- learning_rate: 2e-05
+- train_batch_size: 8
+- eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 2
-- total_train_batch_size: 8
+- total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 150
+- lr_scheduler_warmup_steps: 75
 - num_epochs: 5
 - mixed_precision_training: Native AMP
 
@@ -56,11 +56,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss | Wer    |
 |:-------------:|:------:|:----:|:---------------:|:------:|
-| 0.3794        | 0.9966 | 146  | 0.1997          | 0.1031 |
-| 0.085         | 2.0    | 293  | 0.1471          | 0.0843 |
-| 0.0156        | 2.9966 | 439  | 0.1455          | 0.0690 |
-| 0.0052        | 4.0    | 586  | 0.1518          | 0.0681 |
-| 0.001         | 4.9829 | 730  | 0.1558          | 0.0674 |
+| 0.3839        | 0.9932 | 73   | 0.1968          | 0.0975 |
+| 0.0763        | 2.0    | 147  | 0.1418          | 0.0879 |
+| 0.017         | 2.9932 | 220  | 0.1410          | 0.1200 |
+| 0.0058        | 4.0    | 294  | 0.1404          | 0.0645 |
+| 0.0014        | 4.9660 | 365  | 0.1396          | 0.0647 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:239d0e2a69e501c4a0a89f04f52988f6a2b054de366d8dd7ea7231c5e05285a0
+oid sha256:ae2b85d11442b331ad7d2f7e0c2c019cc41d1ca1491814072d30dbcda008a042
 size 3055544304

runs/Oct27_12-28-17_dc2bdd442fe7/events.out.tfevents.1730035281.dc2bdd442fe7.334.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:707fe0b001d22be1670b859103cdf97bc75b4731399f2282c72c53cfa3fc7eab
+size 406