Merge branch 'main' of https://huggingface.co./KBLab/wav2vec2-large-xlsr-53-swedish into main
Browse files
README.md
CHANGED
@@ -25,10 +25,10 @@ model-index:
|
|
25 |
metrics:
|
26 |
- name: Test WER
|
27 |
type: wer
|
28 |
-
value:
|
29 |
- name: Test CER
|
30 |
type: cer
|
31 |
-
value: 5.
|
32 |
---
|
33 |
|
34 |
# Wav2Vec2-Large-XLSR-53-Swedish
|
@@ -93,7 +93,7 @@ processor = Wav2Vec2Processor.from_pretrained("KBLab/wav2vec2-large-xlsr-53-swed
|
|
93 |
model = Wav2Vec2ForCTC.from_pretrained("KBLab/wav2vec2-large-xlsr-53-swedish")
|
94 |
model.to("cuda")
|
95 |
|
96 |
-
chars_to_ignore_regex = '[
|
97 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|
98 |
|
99 |
# Preprocessing the datasets.
|
@@ -126,8 +126,8 @@ print("WER: {:2f}".format(100 * wer.compute(predictions=result["pred_strings"],
|
|
126 |
print("CER: {:2f}".format(100 * wer.compute(predictions=[" ".join(list(entry)) for entry in result["pred_strings"]], references=[" ".join(list(entry)) for entry in result["sentence"]])))
|
127 |
```
|
128 |
|
129 |
-
**WER**:
|
130 |
-
**CER**: 5.
|
131 |
|
132 |
## Training
|
133 |
|
|
|
25 |
metrics:
|
26 |
- name: Test WER
|
27 |
type: wer
|
28 |
+
value: 16.943492
|
29 |
- name: Test CER
|
30 |
type: cer
|
31 |
+
value: 5.509278
|
32 |
---
|
33 |
|
34 |
# Wav2Vec2-Large-XLSR-53-Swedish
|
|
|
93 |
model = Wav2Vec2ForCTC.from_pretrained("KBLab/wav2vec2-large-xlsr-53-swedish")
|
94 |
model.to("cuda")
|
95 |
|
96 |
+
chars_to_ignore_regex = '[,?.!\\\\\\\\\\\\\\\\-;:"“]'
|
97 |
resampler = torchaudio.transforms.Resample(48_000, 16_000)
|
98 |
|
99 |
# Preprocessing the datasets.
|
|
|
126 |
print("CER: {:2f}".format(100 * wer.compute(predictions=[" ".join(list(entry)) for entry in result["pred_strings"]], references=[" ".join(list(entry)) for entry in result["sentence"]])))
|
127 |
```
|
128 |
|
129 |
+
**WER**: 16.94%
|
130 |
+
**CER**: 5.51%
|
131 |
|
132 |
## Training
|
133 |
|