CoolFace
Modelpublic

pierreguillou/whisper-medium-portuguese

sourceHugging Faceapache-2.0updated 4y agoView on Hugging Face
32likes160downloads
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. -->

Portuguese Medium Whisper

This model is a fine-tuned version of openai/whisper-medium on the commonvoice11_0 dataset. It achieves the following results on the evaluation set:

  • —Loss: 0.2628
  • —Wer: 6.5987

Blog post

All information about this model in this blog post: Speech-to-Text & IA | Transcreva qualquer áudio para o português com o Whisper (OpenAI)... sem nenhum custo!.

New SOTA

The Normalized WER in the OpenAI Whisper article with the Common Voice 9.0 test dataset is 8.1.

As this test dataset is similar to the Common Voice 11.0 test dataset used to evaluate our model (WER and WER Norm), it means that our Portuguese Medium Whisper is better than the [Medium Whisper](https://huggingface.co/openai/whisper-medium) model at transcribing audios Portuguese in text (and even better than the Whisper Large that has a WER Norm of 7.1!).

OpenAI results with Whisper Medium and Test dataset of Commons Voice 9.0

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • —learning_rate: 9e-06
  • —trainbatchsize: 32
  • —evalbatchsize: 16
  • —seed: 42
  • —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • —lrschedulertype: linear
  • —lrschedulerwarmup_steps: 500
  • —training_steps: 6000
  • —mixedprecisiontraining: Native AMP

Training results

Training LossEpochStepValidation LossWer
0.03332.0715000.20736.9770
0.00615.0530000.26286.5987
0.00078.0345000.29606.6979
0.000411.060000.32126.6794

Framework versions

  • —Transformers 4.26.0.dev0
  • —Pytorch 1.13.0+cu117
  • —Datasets 2.7.1.dev0
  • —Tokenizers 0.13.2