From f6f6dd27c5f55fd39200374633b354ea78364517 Mon Sep 17 00:00:00 2001 From: CaioMizerkowski Date: Sat, 20 Apr 2024 11:21:02 -0300 Subject: [PATCH] Update speaker_verification.py Improve performance using onnx and remove the warming: [W:onnxruntime:, session_state.cc:1166 VerifyEachNodeIsAssignedToAnEp] Some nodes were not assigned to the preferred execution providers which may or may not have an negative impact on performance. e.g. ORT explicitly assigns shape related ops to CPU to improve perf. --- pyannote/audio/pipelines/speaker_verification.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pyannote/audio/pipelines/speaker_verification.py b/pyannote/audio/pipelines/speaker_verification.py index 8c4139b6f..c846aa01a 100644 --- a/pyannote/audio/pipelines/speaker_verification.py +++ b/pyannote/audio/pipelines/speaker_verification.py @@ -449,7 +449,8 @@ def to(self, device: torch.device): { "cudnn_conv_algo_search": "DEFAULT", # EXHAUSTIVE / HEURISTIC / DEFAULT }, - ) + ), + "CPUExecutionProvider", ] else: warnings.warn(