mms-1b-bemgen-female-model

This model is a fine-tuned version of facebook/mms-1b-all on the GENBED - BEM dataset. It achieves the following results on the evaluation set:

  • Loss: 0.1970
  • Wer: 0.3638

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
  • train_batch_size: 4
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_steps: 100
  • num_epochs: 30.0
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Wer
6.5936 0.1028 100 0.6191 0.6900
0.4599 0.2055 200 0.2752 0.4553
0.3326 0.3083 300 0.2514 0.4496
0.3384 0.4111 400 0.2390 0.4161
0.2899 0.5139 500 0.2287 0.3987
0.3092 0.6166 600 0.2253 0.4002
0.3262 0.7194 700 0.2193 0.3905
0.2987 0.8222 800 0.2116 0.3934
0.2934 0.9250 900 0.2113 0.3830
0.2932 1.0277 1000 0.2068 0.3802
0.2878 1.1305 1100 0.2107 0.3958
0.2859 1.2333 1200 0.2033 0.3674
0.2782 1.3361 1300 0.2025 0.3749
0.2944 1.4388 1400 0.1995 0.3627
0.2731 1.5416 1500 0.2017 0.3731
0.2735 1.6444 1600 0.1970 0.3636
0.2662 1.7472 1700 0.1981 0.3605
0.2769 1.8499 1800 0.1982 0.3574
0.2748 1.9527 1900 0.2015 0.3852

Framework versions

  • Transformers 4.48.0.dev0
  • Pytorch 2.5.1+cu124
  • Datasets 3.2.0
  • Tokenizers 0.21.0
Downloads last month
0
Safetensors
Model size
965M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Model tree for csikasote/mms-1b-bemgen-female-model

Finetuned
(150)
this model