End of training
Browse files- README.md +23 -12
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -6,6 +6,11 @@ license: apache-2.0
|
|
| 6 |
base_model: openai/whisper-base
|
| 7 |
tags:
|
| 8 |
- generated_from_trainer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
model-index:
|
| 10 |
- name: ./whisper-base-ea-1hrsd
|
| 11 |
results: []
|
|
@@ -18,18 +23,13 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 18 |
|
| 19 |
This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the Afrispeech-200 dataset.
|
| 20 |
It achieves the following results on the evaluation set:
|
| 21 |
-
-
|
| 22 |
-
-
|
| 23 |
-
-
|
| 24 |
-
-
|
| 25 |
-
-
|
| 26 |
-
-
|
| 27 |
-
-
|
| 28 |
-
- eval_runtime: 167.5223
|
| 29 |
-
- eval_samples_per_second: 3.779
|
| 30 |
-
- eval_steps_per_second: 0.478
|
| 31 |
-
- epoch: 0.7958
|
| 32 |
-
- step: 300
|
| 33 |
|
| 34 |
## Model description
|
| 35 |
|
|
@@ -58,6 +58,17 @@ The following hyperparameters were used during training:
|
|
| 58 |
- training_steps: 500
|
| 59 |
- mixed_precision_training: Native AMP
|
| 60 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 61 |
### Framework versions
|
| 62 |
|
| 63 |
- Transformers 4.51.3
|
|
|
|
| 6 |
base_model: openai/whisper-base
|
| 7 |
tags:
|
| 8 |
- generated_from_trainer
|
| 9 |
+
metrics:
|
| 10 |
+
- wer
|
| 11 |
+
- precision
|
| 12 |
+
- recall
|
| 13 |
+
- f1
|
| 14 |
model-index:
|
| 15 |
- name: ./whisper-base-ea-1hrsd
|
| 16 |
results: []
|
|
|
|
| 23 |
|
| 24 |
This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the Afrispeech-200 dataset.
|
| 25 |
It achieves the following results on the evaluation set:
|
| 26 |
+
- Loss: 0.7341
|
| 27 |
+
- Wer Ortho: 0.3002
|
| 28 |
+
- Wer: 0.2495
|
| 29 |
+
- Cer: 0.1214
|
| 30 |
+
- Precision: 0.8529
|
| 31 |
+
- Recall: 0.8536
|
| 32 |
+
- F1: 0.8523
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
|
| 34 |
## Model description
|
| 35 |
|
|
|
|
| 58 |
- training_steps: 500
|
| 59 |
- mixed_precision_training: Native AMP
|
| 60 |
|
| 61 |
+
### Training results
|
| 62 |
+
|
| 63 |
+
| Training Loss | Epoch | Step | Validation Loss | Wer Ortho | Wer | Cer | Precision | Recall | F1 |
|
| 64 |
+
|:-------------:|:------:|:----:|:---------------:|:---------:|:------:|:------:|:---------:|:------:|:------:|
|
| 65 |
+
| 0.9512 | 0.3344 | 100 | 0.9677 | 0.3278 | 0.2780 | 0.1320 | 0.8298 | 0.8355 | 0.8319 |
|
| 66 |
+
| 0.7473 | 0.6689 | 200 | 0.8168 | 0.3054 | 0.2477 | 0.1169 | 0.8454 | 0.8479 | 0.8459 |
|
| 67 |
+
| 0.7431 | 1.0033 | 300 | 0.7681 | 0.2922 | 0.2427 | 0.1247 | 0.8537 | 0.8507 | 0.8512 |
|
| 68 |
+
| 0.5364 | 1.3378 | 400 | 0.7507 | 0.2946 | 0.2470 | 0.1240 | 0.8490 | 0.8484 | 0.8476 |
|
| 69 |
+
| 0.5233 | 1.6722 | 500 | 0.7341 | 0.3002 | 0.2495 | 0.1214 | 0.8529 | 0.8536 | 0.8523 |
|
| 70 |
+
|
| 71 |
+
|
| 72 |
### Framework versions
|
| 73 |
|
| 74 |
- Transformers 4.51.3
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 290403936
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:21b9b5d520ce8f7cdff0b68c1986323e32c1f04280595f48a9617a6fb6c86fc3
|
| 3 |
size 290403936
|