magicyoung8 commited on
Commit
0f62d97
·
verified ·
1 Parent(s): 992636d

Update README.md

Browse files

add a line to switch inference mode for the model.

Files changed (1) hide show
  1. README.md +2 -0
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