End of training

README.md

@@ -1,10 +1,9 @@
 ---
+library_name: transformers
 license: apache-2.0
 base_model: openai/whisper-tiny
 tags:
 - generated_from_trainer
-- Whisper Tiny on en-US
-- Whisper Tiny Fine-Tuned
 datasets:
 - PolyAI/minds14
 metrics:
@@ -24,9 +23,7 @@ model-index:
     metrics:
     - name: Wer
       type: wer
-      value: .3327
-language:
-- en
+      value: 0.3284132841328413
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -36,9 +33,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-tiny](https://huggingface.co/openai/whisper-tiny) on the PolyAI/minds14 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.6532
-- Wer Ortho: .3365
-- Wer: .3327
+- Loss: 0.6505
+- Wer Ortho: 0.3327
+- Wer: 0.3284
 
 ## Model description
 
@@ -69,14 +66,14 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch   | Step | Validation Loss | Wer Ortho | Wer     |
-|:-------------:|:-------:|:----:|:---------------:|:---------:|:-------:|
-| 0.0006        | 17.2414 | 500  | 0.6532          | .3365     | .3327   |
+| Training Loss | Epoch   | Step | Validation Loss | Wer Ortho | Wer    |
+|:-------------:|:-------:|:----:|:---------------:|:---------:|:------:|
+| 0.0006        | 17.2414 | 500  | 0.6505          | 0.3327    | 0.3284 |
 
 
 ### Framework versions
 
-- Transformers 4.42.4
-- Pytorch 2.3.1+cu121
-- Datasets 2.20.0
-- Tokenizers 0.19.1
+- Transformers 4.44.2
+- Pytorch 2.4.0+cu121
+- Datasets 2.21.0
+- Tokenizers 0.19.1

config.json

@@ -146,7 +146,7 @@
     50361,
     50362
   ],
-  "torch_dtype": "float32",
+  "torch_dtype": "float16",
   "transformers_version": "4.44.2",
   "use_cache": false,
   "use_weighted_layer_sum": false,

generation_config.json

@@ -244,5 +244,5 @@
     "transcribe": 50359,
     "translate": 50358
   },
-  "transformers_version": "4.42.4"
+  "transformers_version": "4.44.2"
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4d7c5f8044ccc81b5981b5a23a02044ca441a99c331cf7f274bf9b1f9def3eb7
-size 151061672
+oid sha256:7918a0c360a78df95b6d5bf1b4430c9e7e27b5787dc299489d7113880374091c
+size 75540168