Update README.md
Browse filesadd a line to switch inference mode for the model.
README.md
CHANGED
@@ -147,6 +147,8 @@ from nemo.collections.asr.models import SortformerEncLabelModel
|
|
147 |
diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
|
148 |
# load model from Hugging Face model card directly (You need a Hugging Face token)
|
149 |
diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
|
|
|
|
|
150 |
```
|
151 |
|
152 |
### Input Format
|
|
|
147 |
diar_model = SortformerEncLabelModel.restore_from(restore_path="/path/to/diar_sortformer_4spk-v1.nemo", map_location=torch.device('cuda'), strict=False)
|
148 |
# load model from Hugging Face model card directly (You need a Hugging Face token)
|
149 |
diar_model = SortformerEncLabelModel.from_pretrained("nvidia/diar_sortformer_4spk-v1")
|
150 |
+
# switch to inference mode
|
151 |
+
diar_model.eval()
|
152 |
```
|
153 |
|
154 |
### Input Format
|