crossdelenna's picture
Update README.md
5d052c2
|
raw
history blame
1.55 kB
metadata
tags:
  - generated_from_trainer
model-index:
  - name: wav2vec2-base-en-in
    results: []

wav2vec2-base-en-in

This model is a fine-tuned version of beinghorizontal/wav2vec2-base-en-in on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 3.0776
  • Wer: 1.0

Model description

Model for male English Indian accent.

Intended uses & limitations

Intended for male speakers with an Indian accent

Training and evaluation data

Data is all personal based on my Reddit and Twitter chat history Spoken with desktop omnidirectional microphone. Data collection is still in progress.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 0.0001
  • train_batch_size: 16
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_steps: 1000
  • num_epochs: 100
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Wer
3.1239 83.33 500 3.0776 1.0

Framework versions

  • Transformers 4.17.0
  • Pytorch 1.12.0+cu113
  • Datasets 1.18.3
  • Tokenizers 0.12.1