Automatic Speech Recognition
Transformers
TensorBoard
Safetensors
Arabic
whisper
Generated from Trainer
Eval Results (legacy)
Instructions to use deepdml/whisper-base-ar-mix-norm with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use deepdml/whisper-base-ar-mix-norm with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("automatic-speech-recognition", model="deepdml/whisper-base-ar-mix-norm")# Load model directly from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq processor = AutoProcessor.from_pretrained("deepdml/whisper-base-ar-mix-norm") model = AutoModelForSpeechSeq2Seq.from_pretrained("deepdml/whisper-base-ar-mix-norm") - Notebooks
- Google Colab
- Kaggle
End of training
Browse files- README.md +37 -37
- model.safetensors +1 -1
README.md
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
---
|
|
|
|
| 2 |
language:
|
| 3 |
- ar
|
| 4 |
license: apache-2.0
|
|
@@ -6,11 +7,12 @@ base_model: openai/whisper-base
|
|
| 6 |
tags:
|
| 7 |
- generated_from_trainer
|
| 8 |
datasets:
|
| 9 |
-
-
|
| 10 |
-
- fixie-ai/common_voice_17_0
|
| 11 |
- UBC-NLP/Casablanca
|
|
|
|
| 12 |
- google/fleurs
|
| 13 |
-
-
|
|
|
|
| 14 |
metrics:
|
| 15 |
- wer
|
| 16 |
model-index:
|
|
@@ -21,12 +23,13 @@ model-index:
|
|
| 21 |
type: automatic-speech-recognition
|
| 22 |
dataset:
|
| 23 |
name: Common Voice 17.0
|
| 24 |
-
type:
|
| 25 |
metrics:
|
| 26 |
- name: Wer
|
| 27 |
type: wer
|
| 28 |
-
value:
|
| 29 |
---
|
|
|
|
| 30 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 31 |
should probably proofread and complete it, then remove this comment. -->
|
| 32 |
|
|
@@ -34,9 +37,9 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 34 |
|
| 35 |
This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the Common Voice 17.0 dataset.
|
| 36 |
It achieves the following results on the evaluation set:
|
| 37 |
-
- Loss: 0.
|
| 38 |
-
- Wer:
|
| 39 |
-
- Cer:
|
| 40 |
|
| 41 |
## Model description
|
| 42 |
|
|
@@ -59,41 +62,38 @@ The following hyperparameters were used during training:
|
|
| 59 |
- train_batch_size: 64
|
| 60 |
- eval_batch_size: 64
|
| 61 |
- seed: 42
|
| 62 |
-
- optimizer:
|
| 63 |
- lr_scheduler_type: linear
|
| 64 |
- lr_scheduler_warmup_ratio: 0.04
|
| 65 |
-
- training_steps:
|
| 66 |
|
| 67 |
### Training results
|
| 68 |
|
| 69 |
-
| Training Loss | Epoch | Step
|
| 70 |
-
|:-------------:|:------:|:----:|:---------------:|:-------:|:-------:|
|
| 71 |
-
| 0.
|
| 72 |
-
| 0.
|
| 73 |
-
| 0.
|
| 74 |
-
| 0.
|
| 75 |
-
| 0.
|
| 76 |
-
| 0.
|
| 77 |
-
| 0.
|
| 78 |
-
| 0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 79 |
|
| 80 |
|
| 81 |
### Framework versions
|
| 82 |
|
| 83 |
-
- Transformers 4.
|
| 84 |
-
- Pytorch 2.
|
| 85 |
-
- Datasets
|
| 86 |
-
- Tokenizers 0.
|
| 87 |
-
|
| 88 |
-
## Citation
|
| 89 |
-
|
| 90 |
-
Please cite the model using the following BibTeX entry:
|
| 91 |
-
|
| 92 |
-
```bibtex
|
| 93 |
-
@misc{deepdml/whisper-base-ar-mix-norm,
|
| 94 |
-
title={Fine-tuned Whisper base ASR model for speech recognition in Arabic},
|
| 95 |
-
author={Jimenez, David},
|
| 96 |
-
howpublished={\url{https://huggingface.co/deepdml/whisper-base-ar-mix-norm}},
|
| 97 |
-
year={2026}
|
| 98 |
-
}
|
| 99 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
+
library_name: transformers
|
| 3 |
language:
|
| 4 |
- ar
|
| 5 |
license: apache-2.0
|
|
|
|
| 7 |
tags:
|
| 8 |
- generated_from_trainer
|
| 9 |
datasets:
|
| 10 |
+
- deepdml/Tunisian_MSA
|
|
|
|
| 11 |
- UBC-NLP/Casablanca
|
| 12 |
+
- fixie-ai/common_voice_17_0
|
| 13 |
- google/fleurs
|
| 14 |
+
- pain/MASC
|
| 15 |
+
- ymoslem/MediaSpeech
|
| 16 |
metrics:
|
| 17 |
- wer
|
| 18 |
model-index:
|
|
|
|
| 23 |
type: automatic-speech-recognition
|
| 24 |
dataset:
|
| 25 |
name: Common Voice 17.0
|
| 26 |
+
type: deepdml/Tunisian_MSA
|
| 27 |
metrics:
|
| 28 |
- name: Wer
|
| 29 |
type: wer
|
| 30 |
+
value: 39.07383265088779
|
| 31 |
---
|
| 32 |
+
|
| 33 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
| 34 |
should probably proofread and complete it, then remove this comment. -->
|
| 35 |
|
|
|
|
| 37 |
|
| 38 |
This model is a fine-tuned version of [openai/whisper-base](https://huggingface.co/openai/whisper-base) on the Common Voice 17.0 dataset.
|
| 39 |
It achieves the following results on the evaluation set:
|
| 40 |
+
- Loss: 0.4415
|
| 41 |
+
- Wer: 39.0738
|
| 42 |
+
- Cer: 12.5149
|
| 43 |
|
| 44 |
## Model description
|
| 45 |
|
|
|
|
| 62 |
- train_batch_size: 64
|
| 63 |
- eval_batch_size: 64
|
| 64 |
- seed: 42
|
| 65 |
+
- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
|
| 66 |
- lr_scheduler_type: linear
|
| 67 |
- lr_scheduler_warmup_ratio: 0.04
|
| 68 |
+
- training_steps: 18000
|
| 69 |
|
| 70 |
### Training results
|
| 71 |
|
| 72 |
+
| Training Loss | Epoch | Step | Validation Loss | Wer | Cer |
|
| 73 |
+
|:-------------:|:------:|:-----:|:---------------:|:-------:|:-------:|
|
| 74 |
+
| 0.7722 | 0.0556 | 1000 | 0.6243 | 54.6887 | 19.2051 |
|
| 75 |
+
| 0.4875 | 0.1111 | 2000 | 0.5574 | 49.7126 | 16.7685 |
|
| 76 |
+
| 0.2778 | 0.1667 | 3000 | 0.5361 | 47.8269 | 15.8165 |
|
| 77 |
+
| 0.1987 | 0.2222 | 4000 | 0.5172 | 45.5776 | 14.8800 |
|
| 78 |
+
| 0.1477 | 0.2778 | 5000 | 0.5015 | 45.5666 | 14.9556 |
|
| 79 |
+
| 0.1382 | 0.3333 | 6000 | 0.4788 | 43.4751 | 14.1194 |
|
| 80 |
+
| 0.0871 | 0.3889 | 7000 | 0.4689 | 42.8049 | 13.8911 |
|
| 81 |
+
| 0.0769 | 0.4444 | 8000 | 0.4575 | 41.8777 | 13.6289 |
|
| 82 |
+
| 0.082 | 0.5 | 9000 | 0.4542 | 41.3801 | 13.3983 |
|
| 83 |
+
| 0.0873 | 0.5556 | 10000 | 0.4483 | 41.4700 | 13.3972 |
|
| 84 |
+
| 0.0654 | 0.6111 | 11000 | 0.4399 | 40.4179 | 13.1265 |
|
| 85 |
+
| 0.0537 | 0.6667 | 12000 | 0.4434 | 40.2380 | 12.9609 |
|
| 86 |
+
| 0.0806 | 0.7222 | 13000 | 0.4402 | 40.1296 | 12.9885 |
|
| 87 |
+
| 0.0678 | 0.7778 | 14000 | 0.4363 | 39.9589 | 12.8484 |
|
| 88 |
+
| 0.064 | 0.8333 | 15000 | 0.4333 | 39.5476 | 12.7333 |
|
| 89 |
+
| 0.0515 | 0.8889 | 16000 | 0.4360 | 38.8939 | 12.5456 |
|
| 90 |
+
| 0.0725 | 0.9444 | 17000 | 0.4386 | 39.0793 | 12.6691 |
|
| 91 |
+
| 0.0654 | 1.0 | 18000 | 0.4415 | 39.0738 | 12.5149 |
|
| 92 |
|
| 93 |
|
| 94 |
### Framework versions
|
| 95 |
|
| 96 |
+
- Transformers 4.48.0.dev0
|
| 97 |
+
- Pytorch 2.5.1+cu121
|
| 98 |
+
- Datasets 3.6.0
|
| 99 |
+
- Tokenizers 0.21.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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:14f7110dac1328544e9058de3ce0473f1bb888ec39ec8ac8e7d2b5a59cce72d5
|
| 3 |
size 290403936
|