Hugging Face's logo

anggiatm
/
whisper-small-id-v2

Automatic Speech Recognition
Transformers
TensorBoard
Safetensors
whisper
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
whisper-small-id-v2 / README.md
anggiatm's picture
anggiatm
End of training
ddf5b3a verified
preview code
|
raw
history blame contribute delete
1.65 kB
metadata
library_name: transformers
license: apache-2.0
base_model: openai/whisper-small
tags:
  - generated_from_trainer
metrics:
  - wer
model-index:
  - name: whisper-small-id-v2
    results: []

whisper-small-id-v2

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

  • Loss: 0.0832
  • Wer: 8.4163

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: 1e-05
  • 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: 500
  • training_steps: 4000
  • mixed_precision_training: Native AMP

Training results

Training Loss Epoch Step Validation Loss Wer
0.1952 0.3364 1000 0.1546 14.3557
0.1648 0.6727 2000 0.1158 11.4672
0.0563 1.0091 3000 0.0903 9.2774
0.0407 1.3454 4000 0.0832 8.4163

Framework versions

  • Transformers 4.44.2
  • Pytorch 2.4.1+cu121
  • Datasets 3.0.0
  • Tokenizers 0.19.1