End of training

README.md

@@ -3,6 +3,8 @@ license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+metrics:
+- wer
 model-index:
 - name: home_workstation_ASR
   results: []
@@ -14,6 +16,10 @@ should probably proofread and complete it, then remove this comment. -->
 # home_workstation_ASR
 
 This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the None dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.3540
+- Wer Ortho: 20.1592
+- Wer: 15.1297
 
 ## Model description
 
@@ -39,13 +45,16 @@ The following hyperparameters were used during training:
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: constant_with_warmup
 - lr_scheduler_warmup_steps: 50
-- training_steps: 10
+- training_steps: 4000
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer     |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:-------:|
-| No log        | 0.01  | 10   | 3.6352          | 32.4194   | 25.6620 |
+| 0.232         | 0.9   | 1000 | 0.3209          | 21.7751   | 16.8164 |
+| 0.1153        | 1.8   | 2000 | 0.3150          | 20.7647   | 15.7552 |
+| 0.0653        | 2.7   | 3000 | 0.3327          | 20.2443   | 15.2927 |
+| 0.032         | 3.6   | 4000 | 0.3540          | 20.1592   | 15.1297 |
 
 
 ### Framework versions