ErikMkrtchyan/Hy-Generated-audio-data-with-cv20.0
Viewer • Updated • 121k • 77 • 1
How to use ErikMkrtchyan/whisper-small-hy-2 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("automatic-speech-recognition", model="ErikMkrtchyan/whisper-small-hy-2") # Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
processor = AutoProcessor.from_pretrained("ErikMkrtchyan/whisper-small-hy-2")
model = AutoModelForSpeechSeq2Seq.from_pretrained("ErikMkrtchyan/whisper-small-hy-2")This model is a fine-tuned version of openai/whisper-small on the Hy Generated Audio Data dataset. It achieves the following results on the evaluation set:
This model is based on OpenAI's Whisper Small and fine-tuned for Armenian using a combination of real and synthetic audio data. It is designed to transcribe Armenian speech into text.
The dataset contains both real and high-quality synthetic Armenian speech clips.
| Split(1) | # Clips | Duration (hours) |
|---|---|---|
train |
9,300 | 13.53 |
test |
5,818 | 9.16 |
eval |
5,856 | 8.76 |
generated |
100,000 | 113.61 |
generated[2] |
137,419 | 173.76 |
Total duration: ~318 hours
Train set duration(train+generated#1+generated#2: ~300 hours
Test set duration(test+eval) ~18 hours
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Wer |
|---|---|---|---|---|
| 0.0516 | 0.4999 | 7709 | 0.1417 | 33.0858 |
| 0.0366 | 0.9999 | 15418 | 0.1139 | 27.4340 |
| 0.0275 | 1.4998 | 23127 | 0.1057 | 25.0415 |
| 0.0308 | 1.9997 | 30836 | 0.0981 | 23.7545 |
| 0.017 | 2.4997 | 38545 | 0.1016 | 23.2408 |
| 0.019 | 2.9996 | 46254 | 0.0999 | 22.7854 |