perf: use heuristic instead of exhaustiv conv algo search

This commit is contained in:
marty1885
2023-08-09 16:04:59 +08:00
parent 0d2be1e19e
commit e0f71939b9

View File

@@ -41,7 +41,7 @@ class PiperVoice:
sess_options=onnxruntime.SessionOptions(),
providers=["CPUExecutionProvider"]
if not use_cuda
else ["CUDAExecutionProvider"],
else [("CUDAExecutionProvider", {"cudnn_conv_algo_search": "HEURISTIC"})],
),
)