patrickvonplaten
commited on
Commit
·
cf31dbf
1
Parent(s):
5af67a2
Update README.md
Browse files
README.md
CHANGED
@@ -32,11 +32,11 @@ Wav2Vec2 Conformer with relative position embeddings, pretrained and fine-tuned
|
|
32 |
|
33 |
[Paper (TODO)](https://arxiv.org/abs/2006.11477)
|
34 |
|
35 |
-
Authors:
|
36 |
|
37 |
**Abstract**
|
38 |
|
39 |
-
|
40 |
|
41 |
The original model can be found under https://github.com/pytorch/fairseq/tree/master/examples/wav2vec#wav2vec-20.
|
42 |
|
@@ -51,8 +51,8 @@ To transcribe audio files the model can be used as a standalone acoustic model a
|
|
51 |
import torch
|
52 |
|
53 |
# load model and processor
|
54 |
-
processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-large-960h-
|
55 |
-
model = Wav2Vec2ConformerForCTC.from_pretrained("facebook/wav2vec2-large-960h-
|
56 |
|
57 |
# load dummy dataset and read soundfiles
|
58 |
ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
|
@@ -70,7 +70,7 @@ To transcribe audio files the model can be used as a standalone acoustic model a
|
|
70 |
|
71 |
## Evaluation
|
72 |
|
73 |
-
This code snippet shows how to evaluate **facebook/wav2vec2-large-960h-
|
74 |
|
75 |
```python
|
76 |
from datasets import load_dataset
|
|
|
32 |
|
33 |
[Paper (TODO)](https://arxiv.org/abs/2006.11477)
|
34 |
|
35 |
+
Authors: ...
|
36 |
|
37 |
**Abstract**
|
38 |
|
39 |
+
...
|
40 |
|
41 |
The original model can be found under https://github.com/pytorch/fairseq/tree/master/examples/wav2vec#wav2vec-20.
|
42 |
|
|
|
51 |
import torch
|
52 |
|
53 |
# load model and processor
|
54 |
+
processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-conformer-rel-pos-large-960h-ft")
|
55 |
+
model = Wav2Vec2ConformerForCTC.from_pretrained("facebook/wav2vec2-conformer-rel-pos-large-960h-ft")
|
56 |
|
57 |
# load dummy dataset and read soundfiles
|
58 |
ds = load_dataset("patrickvonplaten/librispeech_asr_dummy", "clean", split="validation")
|
|
|
70 |
|
71 |
## Evaluation
|
72 |
|
73 |
+
This code snippet shows how to evaluate **facebook/wav2vec2-conformer-rel-pos-large-960h-ft** on LibriSpeech's "clean" and "other" test data.
|
74 |
|
75 |
```python
|
76 |
from datasets import load_dataset
|