CoolFace
Modelpublic

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

sourceHugging Faceapache-2.0updated 5y agoView on Hugging Face
0likes26downloads
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-v2

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.5328
  • —Wer: 0.4596

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-v2 --dataset mozilla-foundation/commonvoice80 --config hsb --split test --log_outputs

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

Upper Sorbian (hsb) not found 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.59793.231003.56021.0
3.3036.452003.22381.0
3.20349.683003.20020.9888
2.798612.94001.24080.9210
1.386916.135000.79730.7462
1.022819.356000.67220.6788
0.831122.587000.61000.6150
0.71725.818000.62360.6013
0.626429.039000.60310.5575
0.549432.2610000.56560.5309
0.478135.4811000.52890.4996
0.431138.7112000.53750.4768
0.390241.9413000.52460.4703
0.350845.1614000.53820.4696
0.319948.3915000.53280.4596

Framework versions

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