SPEAK-ASR/youtube-sinhala-asr-preprocessed
Viewer • Updated • 2.99k • 15
How to use SPEAK-ASR/whisper-si-exp-4 with PEFT:
from peft import PeftModel
from transformers import AutoModelForSeq2SeqLM
base_model = AutoModelForSeq2SeqLM.from_pretrained("openai/whisper-small")
model = PeftModel.from_pretrained(base_model, "SPEAK-ASR/whisper-si-exp-4")How to use SPEAK-ASR/whisper-si-exp-4 with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("SPEAK-ASR/whisper-si-exp-4", dtype="auto")This model is a fine-tuned version of openai/whisper-small on the Whisper Small - Sinhala ASR Fine-Tuned dataset. It achieves the following results on the evaluation set:
SPEAK-ASR/whisper-small-si-openslr_plus_youtube_dataMore information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss |
|---|---|---|---|
| 0.3137 | 0.4254 | 1000 | 0.2489 |
| 0.2126 | 0.8507 | 2000 | 0.1943 |
| 0.1506 | 1.2761 | 3000 | 0.1737 |
| 0.1835 | 1.7014 | 4000 | 0.1607 |
| 0.1879 | 2.1268 | 5000 | 0.1538 |
| 0.1485 | 2.5521 | 6000 | 0.1494 |
| 0.1473 | 2.9775 | 7000 | 0.1465 |
Base model
openai/whisper-small