[cli] add seed to prevent randomization of results (#270)

wenet-e2e · Jan 31, 2024 · 5efdd48 · 5efdd48
1 parent 8a80d2f
commit 5efdd48
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/wespeaker/cli/speaker.py b/wespeaker/cli/speaker.py
@@ -32,11 +32,14 @@
 from wespeaker.diar.spectral_clusterer import cluster
 from wespeaker.diar.extract_emb import subsegment
 from wespeaker.diar.make_rttm import merge_segments
+from wespeaker.utils.utils import set_seed
 
 
 class Speaker:
 
     def __init__(self, model_dir: str):
+        set_seed()
+
         config_path = os.path.join(model_dir, 'config.yaml')
         model_path = os.path.join(model_dir, 'avg_model.pt')
         with open(config_path, 'r') as fin: