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.4521
-- Wer: 22.8402
-- Cer: 12.5097
 
 ## Model description
 
@@ -38,35 +32,21 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1.5e-05
+- learning_rate: 5e-06
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
 - gradient_accumulation_steps: 2
 - total_train_batch_size: 16
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
-- lr_scheduler_type: linear
+- lr_scheduler_type: cosine
 - 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.3437        | 1.0   | 278  | 0.3292          | 23.6377 | 13.0135 |
-| 0.1749        | 2.0   | 556  | 0.3337          | 22.6309 | 12.1392 |
-| 0.0823        | 3.0   | 834  | 0.3635          | 23.1891 | 12.9637 |
-| 0.0397        | 4.0   | 1112 | 0.3959          | 23.1559 | 12.5382 |
-| 0.0202        | 5.0   | 1390 | 0.4146          | 24.8771 | 13.9211 |
-| 0.0097        | 6.0   | 1668 | 0.4196          | 22.1192 | 12.0233 |
-| 0.0038        | 7.0   | 1946 | 0.4424          | 22.2455 | 12.3308 |
-| 0.0018        | 8.0   | 2224 | 0.4521          | 22.8402 | 12.5097 |
-
-
 ### Framework versions
 
-- Transformers 4.38.1
+- Transformers 4.38.2
 - Pytorch 2.2.1+cu121
-- Datasets 2.17.2.dev0
+- Datasets 2.18.0
 - Tokenizers 0.15.2

config.json

@@ -45,8 +45,8 @@
   "scale_embedding": false,
   "suppress_tokens": [],
   "torch_dtype": "float32",
-  "transformers_version": "4.38.1",
-  "use_cache": true,
+  "transformers_version": "4.38.2",
+  "use_cache": false,
   "use_weighted_layer_sum": false,
   "vocab_size": 51865
 }

generation_config.json

@@ -313,5 +313,5 @@
     "transcribe": 50359,
     "translate": 50358
   },
-  "transformers_version": "4.38.1"
+  "transformers_version": "4.38.2"
 }

model-00001-of-00002.safetensors

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

model-00002-of-00002.safetensors

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

training_args.bin

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