Debug(infer_tool.py) debug feature index

This commit is contained in:
YuriHead 2023-06-04 00:37:38 +08:00 committed by GitHub
parent 0067f8120b
commit 4f9e440f04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -232,7 +232,7 @@ class Svc(object):
speaker_id = speaker
feature_index = self.cluster_model[speaker_id]
feat_np = c.transpose(0,1).cpu().numpy()
if self.big_npy is not None or self.now_spk_id != speaker_id:
if self.big_npy is None or self.now_spk_id != speaker_id:
self.big_npy = feature_index.reconstruct_n(0, feature_index.ntotal)
self.now_spk_id = speaker_id
print("starting feature retrieval...")