Automatic Speech Recognition
NeMo
Safetensors
PyTorch
sortformer
speaker-diarization
speaker-recognition
speech
audio
Transformer
FastConformer
Conformer
NEST
NeMo
Eval Results (legacy)
Instructions to use nvidia/diar_sortformer_4spk-v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use nvidia/diar_sortformer_4spk-v1 with NeMo:
import nemo.collections.asr as nemo_asr asr_model = nemo_asr.models.ASRModel.from_pretrained("nvidia/diar_sortformer_4spk-v1") transcriptions = asr_model.transcribe(["file.wav"]) - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -209,7 +209,7 @@ The model can be trained using this [example script](https://github.com/NVIDIA/N
|
|
| 209 |
|
| 210 |
To evaluate Sortformer diarizer and save diarization results in RTTM format, use the inference [example script](https://github.com/NVIDIA/NeMo/blob/main/examples/speaker_tasks/diarization/neural_diarizer/e2e_diarize_speech.py):
|
| 211 |
```bash
|
| 212 |
-
python
|
| 213 |
model_path="/path/to/diar_sortformer_4spk-v1.nemo" \
|
| 214 |
manifest_filepath="/path/to/multispeaker_manifest_with_reference_rttms.json" \
|
| 215 |
collar=COLLAR \
|
|
@@ -218,7 +218,7 @@ python [NEMO_GIT_FOLDER]/examples/speaker_tasks/diarization/neural_diarizer/e2e_
|
|
| 218 |
|
| 219 |
You can provide the post-processing YAML configs from [`post_processing` folder](https://github.com/NVIDIA/NeMo/tree/main/examples/speaker_tasks/diarization/conf/post_processing) to reproduce the optimized post-processing algorithm for each development dataset:
|
| 220 |
```bash
|
| 221 |
-
python
|
| 222 |
model_path="/path/to/diar_sortformer_4spk-v1.nemo" \
|
| 223 |
manifest_filepath="/path/to/multispeaker_manifest_with_reference_rttms.json" \
|
| 224 |
collar=COLLAR \
|
|
|
|
| 209 |
|
| 210 |
To evaluate Sortformer diarizer and save diarization results in RTTM format, use the inference [example script](https://github.com/NVIDIA/NeMo/blob/main/examples/speaker_tasks/diarization/neural_diarizer/e2e_diarize_speech.py):
|
| 211 |
```bash
|
| 212 |
+
python ${NEMO_GIT_FOLDER}/examples/speaker_tasks/diarization/neural_diarizer/e2e_diarize_speech.py
|
| 213 |
model_path="/path/to/diar_sortformer_4spk-v1.nemo" \
|
| 214 |
manifest_filepath="/path/to/multispeaker_manifest_with_reference_rttms.json" \
|
| 215 |
collar=COLLAR \
|
|
|
|
| 218 |
|
| 219 |
You can provide the post-processing YAML configs from [`post_processing` folder](https://github.com/NVIDIA/NeMo/tree/main/examples/speaker_tasks/diarization/conf/post_processing) to reproduce the optimized post-processing algorithm for each development dataset:
|
| 220 |
```bash
|
| 221 |
+
python ${NEMO_GIT_FOLDER}/examples/speaker_tasks/diarization/neural_diarizer/e2e_diarize_speech.py \
|
| 222 |
model_path="/path/to/diar_sortformer_4spk-v1.nemo" \
|
| 223 |
manifest_filepath="/path/to/multispeaker_manifest_with_reference_rttms.json" \
|
| 224 |
collar=COLLAR \
|