CoolFace
Modelpublic

dmusingu/luganda_wav2vec2_ctc_tokenizer

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes14downloads
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. -->

lugandawav2vec2ctc_tokenizer

This model is a fine-tuned version of facebook/wav2vec2-base on the commonvoice7_0 dataset. It achieves the following results on the evaluation set:

  • Loss: 0.5588
  • Wer: 0.5609

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0001
  • trainbatchsize: 32
  • evalbatchsize: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 1000
  • num_epochs: 30
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
4.13652.45001.95981.0
0.56954.8110000.58530.7329
0.1767.2115000.53810.6747
0.08459.6220000.51280.6270
0.042412.0225000.46510.6014
0.012714.4230000.53950.6049
-0.006316.8335000.51690.5842
-0.021219.2340000.49900.5833
-0.033621.6345000.53180.5680
-0.042424.0450000.54650.5702
-0.049526.4455000.55410.5637
-0.056528.8560000.55880.5609

Framework versions

  • Transformers 4.37.2
  • Pytorch 2.1.0+cu121
  • Datasets 2.17.1
  • Tokenizers 0.15.2