nits and minor improvements (#384)

* Update utils.py Sorting once is sufficient * Update utils.py, nit * Update executor.py
wenet-e2e · Nov 14, 2024 · 4fdc23a · 4fdc23a
1 parent e9bbf73
commit 4fdc23a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/wespeaker/utils/executor.py b/wespeaker/utils/executor.py
@@ -57,7 +57,6 @@ def run_epoch(dataloader, epoch_iter, model, criterion, optimizer, scheduler,
                 features = spec_aug(features,
                                     **configs['dataset_args']['spec_aug_args'])
 
-        with torch.cuda.amp.autocast(enabled=configs['enable_amp']):
             outputs = model(features)  # (embed_a,embed_b) in most cases
             embeds = outputs[-1] if isinstance(outputs, tuple) else outputs
             outputs = model.module.projection(embeds, targets)

diff --git a/wespeaker/utils/utils.py b/wespeaker/utils/utils.py
@@ -75,10 +75,9 @@ def set_seed(seed=42):
 
 def spk2id(utt_spk_list):
     _, spk_list = zip(*utt_spk_list)
-    spk_list = sorted(list(set(spk_list)))  # remove overlap and sort
+    spk_list = sorted(set(spk_list))  # remove overlap and sort
 
     spk2id_dict = {}
-    spk_list.sort()
     for i, spk in enumerate(spk_list):
         spk2id_dict[spk] = i
     return spk2id_dict