Skip to content

Commit

Permalink
Fix speaker extraction
Browse files Browse the repository at this point in the history
  • Loading branch information
Unal Ege Gaznepoglu committed Dec 20, 2023
1 parent 28e6c58 commit 33c3dbf
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -78,7 +78,7 @@ def extract_speakers(self, dataset_path, dataset_name=None, emb_level=None):
utts = [x[1] for x in returns]
utts = list(np.concatenate(utts))
else:
vectors, utts = extraction_job([wav_scp, self.extractors, 0, self.devices[0], self.model_hparams, 0])
vectors, utts = extraction_job([wav_scp, self.extractors[0], 0, self.devices[0], self.model_hparams, 0])
vectors = torch.stack(vectors, dim=0)

speakers = [utt2spk[utt] for utt in utts]
Expand Down

0 comments on commit 33c3dbf

Please sign in to comment.