jpterry commited on
Commit
a59c129
·
1 Parent(s): 2b5e23b

num threads = 1

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def load_model(model: str, activation: bool=True):
45
  model += '_w_activation'
46
 
47
  sess_options = ort.SessionOptions()
48
- sess_options.intra_op_num_threads = 0
49
  ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model), sess_options)
50
 
51
  return ort_session
 
45
  model += '_w_activation'
46
 
47
  sess_options = ort.SessionOptions()
48
+ sess_options.intra_op_num_threads = 1
49
  ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model), sess_options)
50
 
51
  return ort_session