End of training

README.md

@@ -18,8 +18,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-medium.en](https://huggingface.co/openai/whisper-medium.en) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5759
-- Wer: 0.1630
+- Loss: 1.0557
+- Wer: 0.1819
 
 ## Model description
 
@@ -38,9 +38,9 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 0.0001
-- train_batch_size: 8
-- eval_batch_size: 8
+- learning_rate: 1e-06
+- train_batch_size: 16
+- eval_batch_size: 4
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
@@ -50,18 +50,18 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch  | Step | Validation Loss | Wer    |
-|:-------------:|:------:|:----:|:---------------:|:------:|
-| 1.6889        | 0.5682 | 25   | 1.0935          | 0.2250 |
-| 0.5293        | 1.1364 | 50   | 0.7940          | 0.4637 |
-| 0.2548        | 1.7045 | 75   | 0.9126          | 0.2702 |
-| 0.1804        | 2.2727 | 100  | 0.6907          | 0.1767 |
-| 0.1173        | 2.8409 | 125  | 0.7136          | 0.1903 |
-| 0.0631        | 3.4091 | 150  | 0.7046          | 0.1966 |
-| 0.046         | 3.9773 | 175  | 0.6465          | 0.1872 |
-| 0.029         | 4.5455 | 200  | 0.6110          | 0.3502 |
-| 0.0107        | 5.1136 | 225  | 0.6141          | 0.1661 |
-| 0.006         | 5.6818 | 250  | 0.5759          | 0.1630 |
+| Training Loss | Epoch   | Step | Validation Loss | Wer    |
+|:-------------:|:-------:|:----:|:---------------:|:------:|
+| 2.466         | 1.1364  | 25   | 2.3416          | 0.3281 |
+| 1.7788        | 2.2727  | 50   | 1.5774          | 0.2671 |
+| 1.2151        | 3.4091  | 75   | 1.2743          | 0.2429 |
+| 0.9613        | 4.5455  | 100  | 1.1783          | 0.2135 |
+| 0.8534        | 5.6818  | 125  | 1.1291          | 0.2008 |
+| 0.7617        | 6.8182  | 150  | 1.1021          | 0.1924 |
+| 0.77          | 7.9545  | 175  | 1.0804          | 0.1872 |
+| 0.7212        | 9.0909  | 200  | 1.0665          | 0.1819 |
+| 0.7186        | 10.2273 | 225  | 1.0592          | 0.1819 |
+| 0.6799        | 11.3636 | 250  | 1.0557          | 0.1819 |
 
 
 ### Framework versions