CoolFace
Modelpublic

yesj1234/xlsr_5p_ko-en

sourceHugging Faceapache-2.0updated 3y agoView on Hugging Face
0likes5downloads
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. -->

ko-xlsr

This model is a fine-tuned version of facebook/wav2vec2-large-xlsr-53 on the ./SAMPLE_SPEECH.PY - NA dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.4651
  • —Cer: 0.0828

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.0003
  • —trainbatchsize: 2
  • —evalbatchsize: 2
  • —seed: 42
  • —distributed_type: multi-GPU
  • —num_devices: 4
  • —gradientaccumulationsteps: 8
  • —totaltrainbatch_size: 64
  • —totalevalbatch_size: 8
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 1000
  • —num_epochs: 40

Training results

Training LossEpochStepValidation LossCer
6.36733.1715000.61040.1606
0.6566.3330000.43180.1129
0.47299.545000.40100.1028
0.378912.6660000.38670.0977
0.316615.8375000.38570.0936
0.26718.9990000.38910.0912
0.228622.16105000.40740.0910
0.196725.32120000.40790.0878
0.171228.49135000.42890.0865
0.149331.65150000.44560.0850
0.133334.82165000.45730.0843
0.119137.98180000.46330.0833

Framework versions

  • —Transformers 4.34.0
  • —Pytorch 2.0.1+cu117
  • —Datasets 2.14.5
  • —Tokenizers 0.14.0