copilot/OpenSeeFace
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added missing setting for onnxruntime session.

Browse source
This commit is contained in:
Emiliana 2021-10-06 02:07:28 +02:00
parent 50b33bcc48
commit 2eec12225b
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tracker.py
View file

@ -548,7 +548,7 @@ class Tracker():
self.input_name = self.session.get_inputs()[0].name
options = onnxruntime.SessionOptions()
#options.intra_op_num_threads = max(max_threads,4)
options.inter_op_num_threads = 1
options.intra_op_num_threads = 1
options.execution_mode = onnxruntime.ExecutionMode.ORT_SEQUENTIAL
options.graph_optimization_level = onnxruntime.GraphOptimizationLevel.ORT_ENABLE_ALL