trying to set number of onnx threads
Browse files
app.py
CHANGED
|
@@ -47,6 +47,7 @@ def load_model(model: str, activation: bool=True):
|
|
| 47 |
print(model)
|
| 48 |
|
| 49 |
ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model))
|
|
|
|
| 50 |
|
| 51 |
return ort_session
|
| 52 |
|
|
|
|
| 47 |
print(model)
|
| 48 |
|
| 49 |
ort_session = ort.InferenceSession(model_path + '%s.onnx' % (model))
|
| 50 |
+
ort_session.intra_op_num_threads = 1
|
| 51 |
|
| 52 |
return ort_session
|
| 53 |
|