TingChen-ppmc/Shanghai_Dialect_Conversational_Speech_Corpus
Viewer • Updated • 3.79k • 150 • 12
How to use wangyuzhe/whisper-small-dv with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("automatic-speech-recognition", model="wangyuzhe/whisper-small-dv") # Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
processor = AutoProcessor.from_pretrained("wangyuzhe/whisper-small-dv")
model = AutoModelForSpeechSeq2Seq.from_pretrained("wangyuzhe/whisper-small-dv", device_map="auto")This model is a fine-tuned version of openai/whisper-small on the Shanghai Dialect Conversational Speech Corpus 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 Ortho | Cer |
|---|---|---|---|---|---|
| 0.9990 | 0.6024 | 100 | 0.9789 | 53.6472 | 53.2233 |
| 0.5426 | 1.2048 | 200 | 0.7389 | 41.0817 | 40.5007 |
| 0.4840 | 1.8072 | 300 | 0.6436 | 38.6281 | 38.0243 |
| 0.2562 | 2.4096 | 400 | 0.6285 | 34.8212 | 34.2246 |
| 0.2487 | 3.0120 | 500 | 0.6211 | 33.3881 | 32.7486 |
Base model
openai/whisper-small