jsnfly
commited on
Commit
•
9024e1e
1
Parent(s):
9da7e2a
increase beam size
Browse files- README.md +2 -2
- eval.py +1 -0
- log_mozilla-foundation_common_voice_7_0_de_test_predictions.txt +0 -0
README.md
CHANGED
|
@@ -21,10 +21,10 @@ model-index:
|
|
| 21 |
metrics:
|
| 22 |
- name: Test WER
|
| 23 |
type: wer
|
| 24 |
-
value:
|
| 25 |
- name: Test CER
|
| 26 |
type: cer
|
| 27 |
-
value:
|
| 28 |
---
|
| 29 |
|
| 30 |
# Wav2Vec2-Large-XLSR-53-German-GPT2
|
|
|
|
| 21 |
metrics:
|
| 22 |
- name: Test WER
|
| 23 |
type: wer
|
| 24 |
+
value: 10.02
|
| 25 |
- name: Test CER
|
| 26 |
type: cer
|
| 27 |
+
value: 4.7
|
| 28 |
---
|
| 29 |
|
| 30 |
# Wav2Vec2-Large-XLSR-53-German-GPT2
|
eval.py
CHANGED
|
@@ -85,6 +85,7 @@ def main(args):
|
|
| 85 |
|
| 86 |
# load model
|
| 87 |
model = Wav2VecGPT2Model.from_pretrained(args.model_id)
|
|
|
|
| 88 |
|
| 89 |
# load eval pipeline
|
| 90 |
if args.device is None:
|
|
|
|
| 85 |
|
| 86 |
# load model
|
| 87 |
model = Wav2VecGPT2Model.from_pretrained(args.model_id)
|
| 88 |
+
model.config.num_beams = 4
|
| 89 |
|
| 90 |
# load eval pipeline
|
| 91 |
if args.device is None:
|
log_mozilla-foundation_common_voice_7_0_de_test_predictions.txt
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|