CoolFace
Modelpublic

sammy786/wav2vec2-xlsr-dhivehi

sourceHugging Faceapache-2.0updated 5y agoView on Hugging Face
1likes62downloads
Model Card

sammy786/wav2vec2-xlsr-dhivehi

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

  • —Loss: 14.86
  • —Wer: 29.32

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
2004.8838003.1902181.000000
4001.6001000.4978870.726159
6000.9285000.3587810.603892
8000.8679000.3091320.570786
10000.7431000.3091160.552954
12000.7251000.2668390.538378
14000.7862000.2597970.535897
16000.6557000.2456910.517290
18000.6505000.2469570.516204
20000.6855000.2348080.516204
22000.4871000.2284090.507753
24000.4013000.2210870.495968
26000.3593000.2124760.489301
28000.3473000.2048480.487750
30000.3270000.2031630.478756
32000.3371000.2102350.487595
34000.3089000.2014710.491316
36000.2926000.1924370.476120
38000.2896000.1983980.468445
40000.2902000.1934840.467204
42000.2726000.1939990.470150
44000.2667000.1873840.460769
46000.2538000.1872790.476663
48000.2664000.1973950.466817
50000.2580000.1889200.456660
52000.2372000.1807700.457358
54000.2379000.1781490.448287
56000.2326000.1798270.461002
58000.2285000.1821420.445185
60000.2210000.1736190.440688
62000.2195000.1722910.442859
64000.2194000.1733390.430609
66000.2019000.1775520.426423
68000.1990000.1731570.429834
70000.2000000.1665030.423709
72000.1946000.1718120.429834
74000.1921000.1649890.420530
76000.1850000.1683550.418825
78000.1751000.1681280.419290
80000.1735000.1679590.424950
82000.1722000.1736430.414793
84000.1642000.1670200.406342
86000.1708000.1680500.405334
88000.1579000.1642900.396573
90000.1599000.1631880.397426
92000.1517000.1643700.390991
94000.1466000.1650530.392852
96000.1422000.1649390.391844
98000.1483000.1644220.385719
100000.1362000.1665690.385951
102000.1407000.1613770.379594
104000.1333000.1651940.378276
106000.1313000.1643280.369205
108000.1355000.1602540.373236
110000.1211000.1635220.372693

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-dhivehi --dataset mozilla-foundation/common_voice_8_0 --config dv --split test