unanam/mdrama
Viewer • Updated • 3.17k • 13
How to use unanam/whisper-small-ver1 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("automatic-speech-recognition", model="unanam/whisper-small-ver1") # Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
processor = AutoProcessor.from_pretrained("unanam/whisper-small-ver1")
model = AutoModelForSpeechSeq2Seq.from_pretrained("unanam/whisper-small-ver1", device_map="auto")This model is a fine-tuned version of openai/whisper-small on the 사극 드라마 dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Cer |
|---|---|---|---|---|
| 2.9362 | 5.59 | 1000 | 4.0341 | 165.0697 |
| 0.5237 | 11.17 | 2000 | 5.7270 | 88.3566 |
| 0.1343 | 16.76 | 3000 | 7.0163 | 83.8232 |
| 0.0004 | 22.35 | 4000 | 8.2325 | 84.4150 |
Base model
openai/whisper-small