CoolFace
Modelpublic

kaarthu2003/wav2vec2-final-learning-rate-3e-4_batch-size-16

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face
0likes3downloads
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-final-learning-rate-3e-4_batch-size-16

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: inf
  • Wer: 0.3687
  • Cer: 0.0917

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 adamwtorch with betas=(0.9,0.999) and epsilon=1e-08 and optimizerargs=No additional optimizer arguments
  • lrschedulertype: linear
  • lrschedulerwarmup_steps: 400
  • num_epochs: 20
  • mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWerCer
5.68581.0900400inf0.79810.2192
0.72472.1801800inf0.52790.1347
0.51583.27011200inf0.46570.1164
0.42634.36021600inf0.45040.1106
0.34295.45022000inf0.43320.1053
0.29766.54022400inf0.41570.1073
0.25717.63032800inf0.41020.0990
0.21188.72033200inf0.41620.1012
0.18619.81043600inf0.39450.0968
0.159110.90044000inf0.38830.0952
0.136711.99054400inf0.38880.0958
0.114913.07914800inf0.38140.0942
0.098214.16925200inf0.38120.0944
0.082815.25925600inf0.37320.0933
0.073916.34926000inf0.37450.0936
0.067617.43936400inf0.37190.0923
0.05918.52936800inf0.36660.0920
0.05619.61947200inf0.36870.0917

Framework versions

  • Transformers 4.51.1
  • Pytorch 2.1.0+cu118
  • Datasets 3.5.0
  • Tokenizers 0.21.1