Model save

README.md

@@ -1,10 +1,8 @@
 ---
 license: apache-2.0
+base_model: openai/whisper-large-v2
 tags:
 - generated_from_trainer
-base_model: openai/whisper-large-v2
-metrics:
-- wer
 model-index:
 - name: BA_Model_V3
   results: []
@@ -16,10 +14,6 @@ should probably proofread and complete it, then remove this comment. -->
 # BA_Model_V3
 
 This model is a fine-tuned version of [openai/whisper-large-v2](https://huggingface.co/openai/whisper-large-v2) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Loss: 0.5502
-- Wer: 24.6079
-- Cer: 13.4358
 
 ## Model description
 
@@ -38,7 +32,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
+- learning_rate: 1.5e-05
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
@@ -46,31 +40,10 @@ The following hyperparameters were used during training:
 - total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- lr_scheduler_warmup_ratio: 0.2
-- num_epochs: 15
+- lr_scheduler_warmup_ratio: 0.05
+- num_epochs: 8
 - mixed_precision_training: Native AMP
 
-### Training results
-
-| Training Loss | Epoch | Step | Validation Loss | Wer     | Cer     |
-|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
-| 0.3329        | 1.0   | 278  | 0.3232          | 21.6707 | 12.0794 |
-| 0.2364        | 2.0   | 556  | 0.3756          | 23.9367 | 13.5422 |
-| 0.1902        | 3.0   | 834  | 0.4029          | 38.9055 | 23.2284 |
-| 0.1324        | 4.0   | 1112 | 0.4090          | 24.9568 | 13.9016 |
-| 0.0797        | 5.0   | 1390 | 0.4468          | 28.4789 | 15.9201 |
-| 0.0505        | 6.0   | 1668 | 0.4519          | 26.8873 | 14.9458 |
-| 0.0317        | 7.0   | 1946 | 0.4599          | 32.2435 | 18.9045 |
-| 0.0198        | 8.0   | 2224 | 0.4814          | 25.7675 | 14.1053 |
-| 0.0124        | 9.0   | 2502 | 0.4716          | 26.4620 | 15.1289 |
-| 0.007         | 10.0  | 2780 | 0.5003          | 24.9236 | 13.6666 |
-| 0.0024        | 11.0  | 3058 | 0.5149          | 25.2459 | 14.1990 |
-| 0.0018        | 12.0  | 3336 | 0.5262          | 24.2856 | 13.5232 |
-| 0.0006        | 13.0  | 3614 | 0.5360          | 24.0896 | 13.4676 |
-| 0.0006        | 14.0  | 3892 | 0.5473          | 24.2059 | 13.1781 |
-| 0.0005        | 15.0  | 4170 | 0.5502          | 24.6079 | 13.4358 |
-
-
 ### Framework versions
 
 - Transformers 4.38.1

model-00001-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ef0d762180e7626fab9fc23381c7d56eb8ff93cbce72e6fe7755e4d11722a0c2
+oid sha256:2e2fb7baa2247bb382d1c718bb9af5861bdb30f856fa17786ea090a7cd1dd71e
 size 4992706480

model-00002-of-00002.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5ccfea5a2e9f2a3ada90942bf14b863befdf4b375d1606e4f47b322949710226
+oid sha256:613fe399bd020824967306096f1cf5452953ed84131f9dcbac61fb581d8a8bb4
 size 1180663192

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:80860f7e875f932ebd8c7f7b1bfb9cac42648f1fdb5c451620cab073a47ec884
+oid sha256:4adc7162078682c4051827cd4b01c41268f8295667f1a3dcebce25da8326b7fa
 size 5048