From 085d76f08e4abc98f414bfd56ac1f17b16af6571 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Nekvinda?= Date: Thu, 9 Feb 2023 11:13:02 +0100 Subject: [PATCH] Update utils_vad.py --- utils_vad.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/utils_vad.py b/utils_vad.py index 603f0ca..3626c48 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -13,12 +13,15 @@ class OnnxWrapper(): import numpy as np global np import onnxruntime + + opts = onnxruntime.SessionOptions() + opts.inter_op_num_threads = 1 + opts.intra_op_num_threads = 1 + if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers(): - self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider']) + self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts) else: - self.session = onnxruntime.InferenceSession(path) - self.session.intra_op_num_threads = 1 - self.session.inter_op_num_threads = 1 + self.session = onnxruntime.InferenceSession(path, sess_options=opts) self.reset_states() self.sample_rates = [8000, 16000]