CoolFace
Modelpublic

ogbi/wav2vec2-large-mms-1b-yoruba-test

sourceHugging Facecc-by-nc-4.0updated 3y agoView on Hugging Face
2likes614downloads
Model Card

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

wav2vec2-large-mms-1b-yoruba-test

This model is a fine-tuned version of facebook/mms-1b-all on the commonvoice16_1 dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.6682
  • —Wer: 0.6802

Finetuned by Daniel Ogbuigwe

Model description

This checkpoint is a model fine-tuned for multi-lingual ASR using Facebook's Massive Multilingual Speech project. This checkpoint is based on the Wav2Vec2 architecture and makes use of adapter models to transcribe 1000+ languages. The checkpoint consists of 1 billion parameters and has been fine-tuned from facebook/mms-1b on Yoruba.

Intended uses & limitations

More information needed

Training and evaluation data

Common Voice 16.1 Yoruba data

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 0.001
  • —trainbatchsize: 16
  • —evalbatchsize: 8
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 100
  • —num_epochs: 4
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
4.89230.771000.77100.7413
0.75071.542000.72490.7585
0.70332.313000.71050.7247
0.68883.084000.68290.7229
0.64713.855000.66820.6802

Framework versions

  • —Transformers 4.38.0.dev0
  • —Pytorch 2.1.0+cu121
  • —Datasets 2.16.1
  • —Tokenizers 0.15.0