Update README.md
Browse files
README.md
CHANGED
|
@@ -99,7 +99,7 @@ pipeline_tag: audio-classification
|
|
| 99 |
---
|
| 100 |
|
| 101 |
|
| 102 |
-
# Sortformer Diarizer 4spk
|
| 103 |
|
| 104 |
<style>
|
| 105 |
img {
|
|
@@ -129,6 +129,11 @@ and two feedforward layers with 4 sigmoid outputs for each frame input at the to
|
|
| 129 |
<img src="sortformer-v1-model.png" width="450" />
|
| 130 |
</div>
|
| 131 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 132 |
## NVIDIA NeMo
|
| 133 |
|
| 134 |
To train, fine-tune or perform diarization with Sortformer, you will need to install [NVIDIA NeMo](https://github.com/NVIDIA/NeMo)[5]. We recommend you install it after you've installed Cython and latest PyTorch version.
|
|
|
|
| 99 |
---
|
| 100 |
|
| 101 |
|
| 102 |
+
# Streaming Sortformer Diarizer 4spk v2
|
| 103 |
|
| 104 |
<style>
|
| 105 |
img {
|
|
|
|
| 129 |
<img src="sortformer-v1-model.png" width="450" />
|
| 130 |
</div>
|
| 131 |
|
| 132 |
+
|
| 133 |
+
<div align="center">
|
| 134 |
+
<img src="streaming_sortformer_ani.gif" width="450" />
|
| 135 |
+
</div>
|
| 136 |
+
|
| 137 |
## NVIDIA NeMo
|
| 138 |
|
| 139 |
To train, fine-tune or perform diarization with Sortformer, you will need to install [NVIDIA NeMo](https://github.com/NVIDIA/NeMo)[5]. We recommend you install it after you've installed Cython and latest PyTorch version.
|