mawei/StarPilot
1
0
Fork 0
mirror of https://github.com/firestar5683/StarPilot.git synced 2026-07-01 03:22:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix model runtime on PC (#34210)

Browse Source 
exhaustive
This commit is contained in:
Maxime Desroches
2024-12-11 13:27:39 -08:00
committed by GitHub
parent 1dcdf57395
commit 216ebcaa50
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/modeld/runners/onnxmodel.py
+1 -1
View File
@@ -49,7 +49,7 @@ def create_ort_session(path, fp16_to_fp32):
provider = 'OpenVINOExecutionProvider'
elif 'CUDAExecutionProvider' in ort.get_available_providers() and 'ONNXCPU' not in os.environ:
options.intra_op_num_threads = 2
provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'DEFAULT'})
provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'EXHAUSTIVE'})
else:
options.intra_op_num_threads = 2
options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL