speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

rtf

Browse Source
This commit is contained in:
游雁 2023-03-13 20:00:42 +08:00
parent 08d6f5cc1c
commit 37be12e35e
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
funasr/runtime/python/onnxruntime/rapid_paraformer/utils/utils.py
View File

@ -150,6 +150,7 @@ class OrtInferSession():
def __init__(self, model_file, device_id=-1): def __init__(self, model_file, device_id=-1):
device_id = str(device_id) device_id = str(device_id)
sess_opt = SessionOptions() sess_opt = SessionOptions()
sess_opt.intra_op_num_threads = 4
sess_opt.log_severity_level = 4 sess_opt.log_severity_level = 4
sess_opt.enable_cpu_mem_arena = False sess_opt.enable_cpu_mem_arena = False
sess_opt.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL sess_opt.graph_optimization_level = GraphOptimizationLevel.ORT_ENABLE_ALL