CoolFace
Modelpublic

TheRamsay/wav2vec2-gpt2-enc-dec

sourceHugging Faceupdated 2y 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-gpt2-enc-dec

This model is a fine-tuned version of [](https://huggingface.co/) on the commonvoice17_0 dataset. It achieves the following results on the evaluation set:

  • Loss: 0.3276
  • Wer: 0.8489

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: 1e-05
  • trainbatchsize: 16
  • evalbatchsize: 16
  • seed: 42
  • optimizer: Use adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: linear
  • lrschedulerwarmup_ratio: 0.08
  • num_epochs: 20
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
1.94981.5625500.65480.9324
0.45313.1251000.39590.9020
0.40874.68751500.37350.8894
0.39926.252000.35720.8747
0.37257.81252500.35000.8763
0.36359.3753000.34190.8626
0.364710.93753500.33810.8632
0.3612.54000.33400.8566
0.358814.06254500.33160.8547
0.36215.6255000.32990.8547
0.361317.18755500.32800.8498
0.350518.756000.32760.8489

Framework versions

  • Transformers 4.48.3
  • Pytorch 2.6.0+cu124
  • Datasets 3.2.0
  • Tokenizers 0.21.0