num threads = 1
Browse files
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 =
|
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
|