CoolFace
Modelpublic

Akashpb13/Kabyle_xlsr

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
4likes84downloads
Model Card

Akashpb13/Kabyle_xlsr

This model is a fine-tuned version of facebook/wav2vec2-xls-r-300m on the MOZILLA-FOUNDATION/COMMONVOICE7_0 - hu dataset. It achieves the following results on the evaluation set (which is 10 percent of train data set merged with dev datasets):

  • Loss: 0.159032
  • Wer: 0.187934

Model description

"facebook/wav2vec2-xls-r-300m" was finetuned.

Intended uses & limitations

More information needed

Training and evaluation data

Training data - Common voice Kabyle train.tsv. Only 50,000 records were sampled randomly and trained due to huge size of dataset. Only those points were considered where upvotes were greater than downvotes and duplicates were removed after concatenation of all the datasets given in common voice 7.0

Training procedure

For creating the training dataset, all possible datasets were appended and 90-10 split was used.

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.000096
  • trainbatchsize: 8
  • seed: 13
  • gradientaccumulationsteps: 4
  • lrschedulertype: cosinewithrestarts
  • lrschedulerwarmup_steps: 500
  • num_epochs: 30
  • mixedprecisiontraining: Native AMP

Training results

StepTraining LossValidation LossWer
5007.1998003.1305641.000000
10001.5702000.7180970.734682
15000.8508000.5242270.640532
20000.7122000.4686940.603454
25000.6512000.4138330.573025
30000.6031000.4036800.552847
35000.5533000.3726380.541719
40000.5372000.3537590.531191
45000.5063000.3591090.519601
50000.4796000.3439370.511336
55000.4798000.3382140.503948
60000.4495000.3326000.495221
65000.4392000.3239050.492635
70000.4349000.3104170.484555
75000.4032000.3112470.483262
80000.4015000.2956370.476566
85000.3970000.3013210.471672
90000.3716000.2956390.468440
95000.3707000.2940390.468902
100000.3649000.2911950.468440
105000.3483000.2848980.461098
110000.3501000.2817640.459805
115000.3369000.2910220.461606
120000.3307000.2804670.455234
125000.3225000.2717140.452694
130000.3074000.2895190.455465
135000.3093000.2819220.451217
140000.3048000.2715140.452186
145000.2881000.2868010.446830
150000.2932000.2763090.445399
155000.2898000.2871880.446230
160000.2748000.2864060.441243
165000.2717000.2847540.441520
170000.2625000.2754310.442167
175000.2555000.2765750.439858
180000.2602000.2699110.435425
185000.2506000.2705190.434686
190000.2433000.2676550.437826
195000.2406000.2771090.431731
200000.2372000.2666220.433994
205000.2313000.2730150.428868
210000.2272000.2630240.430161
215000.2204000.2728800.429607
220000.2186000.2723400.426883
225000.2131000.2770660.428407
230000.2050000.2784040.424020
235000.2009000.2708770.418987
240000.1990000.2891200.425821
245000.1961000.2758310.424066
250000.1911000.2828220.421850
255000.1901000.2758200.418248
260000.1788000.2792080.419125
265000.1831000.2714640.419218
270000.1774000.2808690.419680
275000.1718000.2795930.414924
280000.1729000.2769490.417648
285000.1649000.2834910.417786
290000.1648000.2831220.416078
295000.1655000.2819690.415801
300000.1638000.2833190.412753
305000.1535000.2857020.414046
310000.1565000.2850410.412615
315000.1509000.2843360.413723
320000.1518000.2859220.412292
325000.1492000.2894610.412153
330000.1454000.2913220.409567
335000.1456000.2943610.409614
340000.1442000.2906860.409059
345000.1434000.2894740.409844
350000.1435000.2903400.408367
355000.1432000.2895810.407351
360000.1384000.2927820.408736
365000.1379000.2891080.408044
370000.1382000.2921270.407166
375000.1346000.2917970.408413
380000.1398000.2900560.408090
385000.1365000.2911980.408090
390000.1377000.2896960.408044

Framework versions

  • Transformers 4.16.0.dev0
  • Pytorch 1.10.0+cu102
  • Datasets 1.18.3
  • 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 Akashpb13/Kabyle_xlsr --dataset mozilla-foundation/common_voice_8_0 --config kab --split test