return numpy array

Signed-off-by: monica-sekoyan <msekoyan@nvidia.com>
NVIDIA · May 29, 2024 · 418e1f0 · 418e1f0
1 parent 37f0b8c
commit 418e1f0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/asr/models/label_models.py b/nemo/collections/asr/models/label_models.py
@@ -633,7 +633,7 @@ def verify_speakers_batch(self, manifest_filepath1, manifest_filepath2, threshol
         # Decision
         decision = similarity_scores >= threshold
 
-        return decision
+        return decision.cpu().numpy()
 
     @torch.no_grad()
     def batch_inference(self, manifest_filepath, batch_size=32, sample_rate=16000, device='cuda'):