CoolFace
Modelpublic

DrishtiSharma/wav2vec2-large-xls-r-300m-hsb-v1

sourceHugging Faceapache-2.0updated 5y agoView on Hugging Face
0likes65downloads
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-xls-r-300m-hsb-v1

This model is a fine-tuned version of facebook/wav2vec2-xls-r-300m on the MOZILLA-FOUNDATION/COMMONVOICE8_0 - HSB dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.5684
  • —Wer: 0.4402

Evaluation Commands

  1. 1.To evaluate on mozilla-foundation/commonvoice8_0 with test split

python eval.py --modelid DrishtiSharma/wav2vec2-large-xls-r-300m-hsb-v1 --dataset mozilla-foundation/commonvoice80 --config hsb --split test --log_outputs

  1. 1.To evaluate on speech-recognition-community-v2/dev_data

Upper Sorbian language isn't available in speech-recognition-community-v2/dev_data

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 0.00045
  • —trainbatchsize: 16
  • —evalbatchsize: 8
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 32
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 500
  • —num_epochs: 50
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
8.9723.231003.74981.0
3.34016.452003.23201.0
3.20469.683003.17410.9806
2.403112.94001.05790.8996
1.042716.135000.79890.7557
0.74119.356000.64050.6299
0.569922.587000.61290.5928
0.460725.818000.65480.5695
0.382729.039000.62680.5190
0.328232.2610000.59190.5016
0.276435.4811000.59530.4805
0.233538.7112000.57170.4728
0.210641.9413000.56740.4569
0.185945.1614000.56850.4502
0.159248.3915000.56840.4402

Framework versions

  • —Transformers 4.16.1
  • —Pytorch 1.10.0+cu111
  • —Datasets 1.18.2
  • —Tokenizers 0.11.0