End of training

README.md

@@ -6,6 +6,8 @@ license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+metrics:
+- wer
 model-index:
 - name: Whisper-Small Augmented for SEP-28k
   results: []
@@ -17,6 +19,9 @@ should probably proofread and complete it, then remove this comment. -->
 # Whisper-Small Augmented for SEP-28k
 
 This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the SEP-28K dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.7265
+- Wer: 13.7591
 
 ## Model description
 
@@ -42,9 +47,20 @@ The following hyperparameters were used during training:
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- training_steps: 5000
+- training_steps: 5910
 - mixed_precision_training: Native AMP
 
+### Training results
+
+| Training Loss | Epoch   | Step | Validation Loss | Wer     |
+|:-------------:|:-------:|:----:|:---------------:|:-------:|
+| 0.0229        | 5.1020  | 1000 | 0.5061          | 13.5547 |
+| 0.0018        | 10.2041 | 2000 | 0.6259          | 13.6204 |
+| 0.0006        | 15.3061 | 3000 | 0.6754          | 13.6496 |
+| 0.0005        | 20.4082 | 4000 | 0.7105          | 13.7007 |
+| 0.0004        | 25.5102 | 5000 | 0.7265          | 13.7591 |
+
+
 ### Framework versions
 
 - Transformers 4.47.1

model.safetensors

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

runs/Dec28_15-27-04_alien/events.out.tfevents.1735417625.alien.5308.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:05dfb4ab8c505cf995a17a01995e0fc03f570d0c21727fd2b36ca7192ca3af6d
-size 58185
+oid sha256:3dfcd9490257a65baf64474879b52cbd42e9fd605c403f2984652f412a152fb2
+size 58539