đ sammy786/wav2vec2-xlsr-basaa
This model is a fine - tuned version of facebook/wav2vec2-xls-r-1b on the MOZILLA - FOUNDATION/COMMON_VOICE_8_0 - bas dataset. It's designed for automatic speech recognition, offering a solution for converting speech to text with specific performance metrics.
đ Quick Start
This model is a fine - tuned version of facebook/wav2vec2-xls-r-1b on the MOZILLA - FOUNDATION/COMMON_VOICE_8_0 - bas dataset.
It achieves the following results on the evaluation set (which is 10 percent of the train data set merged with other and dev datasets):
⨠Features
- Fine - Tuned Model: Based on "facebook/wav2vec2-xls-r-1b", fine - tuned on the specific bas dataset of MOZILLA - FOUNDATION/COMMON_VOICE_8_0.
- Performance Metrics: Achieves certain loss and WER values on the evaluation set, indicating its effectiveness in speech recognition.
đ Documentation
Model description
"facebook/wav2vec2-xls-r-1b" was finetuned.
Intended uses & limitations
More information needed
Training and evaluation data
Training data -
Common voice Finnish train.tsv, dev.tsv and other.tsv
Training procedure
For creating the train dataset, all possible datasets were appended and a 90 - 10 split was used.
Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 0.000045637994662983496
- train_batch_size: 16
- eval_batch_size: 16
- seed: 13
- gradient_accumulation_steps: 2
- total_train_batch_size: 32
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e - 08
- lr_scheduler_type: cosine_with_restarts
- lr_scheduler_warmup_steps: 500
- num_epochs: 70
- mixed_precision_training: Native AMP
Training results
| Step |
Training Loss |
Validation Loss |
Wer |
| 200 |
6.734100 |
1.605006 |
0.980456 |
| 400 |
1.011200 |
0.364686 |
0.442997 |
| 600 |
0.709300 |
0.300204 |
0.377850 |
| 800 |
0.469800 |
0.315612 |
0.405537 |
| 1000 |
0.464700 |
0.352494 |
0.372964 |
| 1200 |
0.421900 |
0.342533 |
0.368078 |
| 1400 |
0.401900 |
0.351398 |
0.343648 |
| 1600 |
0.429800 |
0.350570 |
0.348534 |
| 1800 |
0.352600 |
0.356601 |
0.358306 |
| 2000 |
0.387200 |
0.355814 |
0.356678 |
| 2200 |
0.362400 |
0.345573 |
0.355049 |
Framework versions
- Transformers 4.16.0.dev0
- Pytorch 1.10.0+cu102
- Datasets 1.17.1.dev0
- Tokenizers 0.10.3
Evaluation Commands
- To evaluate on
mozilla - foundation/common_voice_8_0 with split test
python eval.py --model_id sammy786/wav2vec2-xlsr-basaa --dataset mozilla - foundation/common_voice_8_0 --config bas --split test
đ License
This project is under the Apache - 2.0 license.
| Property |
Details |
| Model Type |
Fine - tuned version of "facebook/wav2vec2-xls-r-1b" on MOZILLA - FOUNDATION/COMMON_VOICE_8_0 - bas dataset |
| Training Data |
Common voice Finnish train.tsv, dev.tsv and other.tsv |
%20--%3e%3cdefs%3e%3cstyle%3e%20.st0%20{%20fill:%20%23061b40;%20}%20.st1%20{%20fill:%20%23306af1;%20}%20.st2%20{%20fill:%20%235ce5cf;%20}%20%3c/style%3e%3c/defs%3e%3cg%3e%3cpath%20class='st0'%20d='M55,10.5h9v3h-9v9h12V7.5h-12v3ZM64,19.5h-6v-3h6v3Z'/%3e%3cpolygon%20class='st0'%20points='69%2016.5%2078%2016.5%2078%2019.5%2069%2019.5%2069%2022.5%2081%2022.5%2081%2013.5%2072%2013.5%2072%2010.5%2081%2010.5%2081%207.5%2069%207.5%2069%2016.5'/%3e%3cpolygon%20class='st0'%20points='95%2010.5%2095%207.5%2083%207.5%2083%2022.5%2095%2022.5%2095%2019.5%2086%2019.5%2086%2016.5%2095%2016.5%2095%2013.5%2086%2013.5%2086%2010.5%2095%2010.5'/%3e%3cpath%20class='st0'%20d='M40,1.5v21h11.6l1.4-1.4v-7.6h0c0,0-1.4-1.5-1.4-1.5l1.4-1.4V2.9l-1.4-1.4h-11.6ZM50,19.5h-7v-6h7v6ZM50,10.5h-7v-6h7v6Z'/%3e%3c/g%3e%3cpath%20class='st1'%20d='M23.1,24L14.7,4.8l-4.9,11.2h3.8l-1.8,4H3.3L12.1,0H2C.9,0,0,.9,0,2v20c0,1.1.9,2,2,2h21.1Z'/%3e%3cpath%20class='st2'%20d='M34,0h-16.8l10.6,24h6.2c1.1,0,2-.9,2-2V2C36,.9,35.1,0,34,0ZM32.5,20h-4V4h4v16Z'/%3e%3c/svg%3e)
