CoolFace
Modelpublic

sammy786/wav2vec2-xlsr-chuvash

sourceHugging Faceapache-2.0updated 5y agoView on Hugging Face
0likes79downloads
Model Card

sammy786/wav2vec2-xlsr-chuvash

This model is a fine-tuned version of facebook/wav2vec2-xls-r-1b on the MOZILLA-FOUNDATION/COMMONVOICE8_0 - cv dataset. It achieves the following results on evaluation set (which is 10 percent of train data set merged with other and dev datasets):

  • Loss: 18.02
  • Wer: 29.22

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 90-10 split was used.

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.000045637994662983496
  • trainbatchsize: 8
  • evalbatchsize: 16
  • seed: 13
  • gradientaccumulationsteps: 4
  • totaltrainbatch_size: 32
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lrschedulertype: cosinewithrestarts
  • lrschedulerwarmup_steps: 500
  • num_epochs: 30
  • mixedprecisiontraining: Native AMP

Training results

StepTraining LossValidation LossWer
2006.5591002.2746871.000000
4001.3461000.5082680.681995
6000.7975000.3911740.572876
8000.5563000.3086200.489283
10000.4358000.2739560.454014
12000.3887000.3110270.499415
14000.3383000.2439770.413874
16000.2940000.2141340.385230
18000.2760000.2459910.397311
20000.2539000.2083240.363016
22000.2336000.2221560.370811
24000.2197000.2026020.364186
26000.2050000.2413390.384451
28000.1760000.2635580.384061
30000.1667000.2117680.333398
32000.1606000.1986770.321512
34000.1546000.2086550.328722
36000.1468000.1880220.317810
38000.1332000.1810830.313133
40000.1342000.1900840.316251
42000.1142000.1930340.312159
44000.1173000.1941220.312354
46000.1123000.1911110.305534
48000.1078000.1859300.302611
50000.1004000.1786250.299883
52000.0998000.1764420.294622
54000.1008000.1779350.294427
56000.0963000.1829030.293843
58000.0942000.1810410.293453
60000.0976000.1798650.290725
62000.0916000.1803270.292868
64000.0931000.1802750.292284

Framework versions

  • Transformers 4.16.0.dev0
  • Pytorch 1.10.0+cu102
  • Datasets 1.17.1.dev0
  • Tokenizers 0.10.3
Evaluation Commands
  1. 1.To evaluate on mozilla-foundation/common_voice_8_0 with split test
bash
python eval.py --model_id sammy786/wav2vec2-xlsr-chuvash --dataset mozilla-foundation/common_voice_8_0 --config cv --split test