CoolFace
Modelpublic

anvitamanne/wav2vec2-train

sourceHugging Faceapache-2.0updated 9mo agoView on Hugging Face
0likes8downloads
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-train

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

  • —Loss: 18.2169
  • —Wer: 1.0277
  • —Cer: 0.8260

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: 16
  • —evalbatchsize: 8
  • —seed: 42
  • —gradientaccumulationsteps: 2
  • —totaltrainbatch_size: 32
  • —optimizer: Use OptimizerNames.ADAMWTORCHFUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • —lrschedulertype: linear
  • —lrschedulerwarmup_ratio: 0.1
  • —num_epochs: 45
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWerCer
1.2383.232616000.59000.61270.1549
1.69976.465132001.17900.87360.2882
6.03339.697748005.97601.00.9996
9.452112.9302640010.74831.00.9800
17.342116.1618800018.21561.02730.8260
17.281319.3943960018.21611.02670.8254
17.338122.62691120018.21581.02780.8258
17.329925.85951280018.21651.02660.8258
17.273129.09101440018.21651.02750.8259
17.343332.32361600018.21611.02780.8259
17.28735.55611760018.21601.02730.8260
17.321738.78871920018.21581.02750.8260
17.296242.02022080018.21691.02770.8260

Framework versions

  • —Transformers 4.57.3
  • —Pytorch 2.9.1+cu128
  • —Datasets 2.21.0
  • —Tokenizers 0.22.2