Model save

README.md

@@ -17,9 +17,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4270
-- Wer: 22.0013
-- Wer Normalized: 18.1723
+- Loss: 0.4987
+- Wer: 22.2974
+- Wer Normalized: 18.5291
 
 ## Model description
 
@@ -38,28 +38,24 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 3.05e-05
-- train_batch_size: 32
-- eval_batch_size: 16
+- learning_rate: 2.05e-05
+- train_batch_size: 16
+- eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
-- lr_scheduler_warmup_steps: 80
-- training_steps: 800
+- lr_scheduler_type: reduce_lr_on_plateau
+- lr_scheduler_warmup_steps: 120
+- training_steps: 2400
 - mixed_precision_training: Native AMP
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Wer     | Wer Normalized |
 |:-------------:|:-----:|:----:|:---------------:|:-------:|:--------------:|
-| 0.6738        | 0.5   | 100  | 0.3943          | 21.7334 | 17.9487        |
-| 0.4816        | 1.01  | 200  | 0.3762          | 20.9203 | 17.1352        |
-| 0.2652        | 1.51  | 300  | 0.3872          | 21.1882 | 17.2827        |
-| 0.2901        | 2.01  | 400  | 0.3912          | 21.4608 | 17.7061        |
-| 0.1408        | 2.51  | 500  | 0.4063          | 21.6112 | 18.0010        |
-| 0.1428        | 3.02  | 600  | 0.4132          | 21.8650 | 18.0201        |
-| 0.0839        | 3.52  | 700  | 0.4252          | 22.3679 | 18.4720        |
-| 0.0906        | 4.02  | 800  | 0.4270          | 22.0013 | 18.1723        |
+| 0.3716        | 1.01  | 400  | 0.3988          | 21.8039 | 17.8916        |
+| 0.2003        | 2.02  | 800  | 0.4440          | 22.3350 | 18.6242        |
+| 0.0571        | 3.02  | 1200 | 0.4960          | 22.5982 | 19.2284        |
+| 0.03          | 4.03  | 1600 | 0.4987          | 22.2974 | 18.5291        |
 
 
 ### Framework versions

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8d345c1fff21d5ac6658c16af92c4dcf9affa9863d8f6e4128991db066ce3f89
+oid sha256:3675778de9704c0376d710d4cddf3a523955c5593bc8fa1091a6b5bee7dca23f
 size 290459230