CoolFace
Modelpublic

BALAKA/wav2vec2-large-xlsr-53-th-swear-words

sourceHugging Facecc-by-sa-4.0updated 3y agoView on Hugging Face
0likes10downloads
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-xlsr-53-th-main

This model is a fine-tuned version of airesearch/wav2vec2-large-xlsr-53-th on the common_voice dataset. It achieves the following results on the evaluation set:

  • —Loss: 1.1340
  • —Wer: 0.4686

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 0.0001
  • —trainbatchsize: 32
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 30
  • —num_epochs: 40
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
5.25243.231003.32221.0
3.29136.452003.18181.0
2.2229.683001.24970.5335
1.155812.94001.07920.5214
0.93416.135001.06630.4986
0.802319.356001.03310.4893
0.704122.587001.08010.4800
0.657625.818001.11230.4886
0.606129.039001.07480.4829
0.564932.2610001.11870.4679
0.571735.4811001.12670.4715
0.526738.7112001.13400.4686

Framework versions

  • —Transformers 4.28.0
  • —Pytorch 2.0.1+cu118
  • —Datasets 1.16.1
  • —Tokenizers 0.13.3