Update README.md with improved frontmatter.
Browse files
README.md
CHANGED
|
@@ -142,26 +142,38 @@ model-index:
|
|
| 142 |
type: wer
|
| 143 |
value: 7.0
|
| 144 |
---
|
| 145 |
-
## Model Overview
|
| 146 |
|
| 147 |
-
|
| 148 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 149 |
|
| 150 |
-
|
| 151 |
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
## NVIDIA Riva: Deployment
|
| 158 |
|
| 159 |
-
|
|
|
|
|
|
|
|
|
|
| 160 |
|
| 161 |
-
|
|
|
|
| 162 |
|
| 163 |
The model is available for use in the NeMo toolkit [3], and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.
|
| 164 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 165 |
### Automatically instantiate the model
|
| 166 |
|
| 167 |
```python
|
|
@@ -189,7 +201,7 @@ python [NEMO_GIT_FOLDER]/examples/asr/transcribe_speech.py
|
|
| 189 |
|
| 190 |
### Input
|
| 191 |
|
| 192 |
-
This model accepts 16000
|
| 193 |
|
| 194 |
### Output
|
| 195 |
|
|
@@ -197,7 +209,7 @@ This model provides transcribed speech as a string for a given audio sample.
|
|
| 197 |
|
| 198 |
## Model Architecture
|
| 199 |
|
| 200 |
-
Conformer-CTC model is a non-autoregressive variant of Conformer model [1] for Automatic Speech Recognition which uses CTC loss/decoding instead of Transducer. You may find more info on the detail of this model here: [Conformer-CTC Model](https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/main/asr/models.html).
|
| 201 |
|
| 202 |
## Training
|
| 203 |
|
|
|
|
| 142 |
type: wer
|
| 143 |
value: 7.0
|
| 144 |
---
|
|
|
|
| 145 |
|
| 146 |
+
<style type="text/css" rel=stylesheet">
|
| 147 |
+
img[src$='#model-badge'] {
|
| 148 |
+
display:inline;
|
| 149 |
+
margin-bottom:0;
|
| 150 |
+
margin-top:0;
|
| 151 |
+
}
|
| 152 |
+
</style>
|
| 153 |
|
| 154 |
+
# NVIDIA Conformer-CTC Large (en-US)
|
| 155 |
|
| 156 |
+
[](#model-architecture)
|
| 157 |
+
| [](#model-architecture)
|
| 158 |
+
| [](#datasets)
|
| 159 |
+
| [](#deployment-with-nvidia-riva)
|
|
|
|
|
|
|
| 160 |
|
| 161 |
+
This model transcribes speech in lowercase English alphabet including spaces and apostrophes, and is trained on several thousand hours of English speech data.
|
| 162 |
+
It is a non-autoregressive "large" variant of Conformer, with around 120 million parameters.
|
| 163 |
+
See the [model architecture](#model-architecture) section and [NeMo documentation](https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/main/asr/models.html#conformer-ctc) for complete architecture details.
|
| 164 |
+
It is also compatible with NVIDIA Riva for [production-grade server deployments](#deployment-with-nvidia-riva).
|
| 165 |
|
| 166 |
+
|
| 167 |
+
## Usage
|
| 168 |
|
| 169 |
The model is available for use in the NeMo toolkit [3], and can be used as a pre-trained checkpoint for inference or for fine-tuning on another dataset.
|
| 170 |
|
| 171 |
+
To train, fine-tune or play with the model you will need to install [NVIDIA NeMo](https://github.com/NVIDIA/NeMo). We recommend you install it after you've installed latest PyTorch version.
|
| 172 |
+
|
| 173 |
+
```
|
| 174 |
+
pip install nemo_toolkit['all']
|
| 175 |
+
```
|
| 176 |
+
|
| 177 |
### Automatically instantiate the model
|
| 178 |
|
| 179 |
```python
|
|
|
|
| 201 |
|
| 202 |
### Input
|
| 203 |
|
| 204 |
+
This model accepts 16000 kHz Mono-channel Audio (wav files) as input.
|
| 205 |
|
| 206 |
### Output
|
| 207 |
|
|
|
|
| 209 |
|
| 210 |
## Model Architecture
|
| 211 |
|
| 212 |
+
Conformer-CTC model is a non-autoregressive variant of Conformer model [1] for Automatic Speech Recognition which uses CTC loss/decoding instead of Transducer. You may find more info on the detail of this model here: [Conformer-CTC Model](https://docs.nvidia.com/deeplearning/nemo/user-guide/docs/en/main/asr/models.html#conformer-ctc).
|
| 213 |
|
| 214 |
## Training
|
| 215 |
|