fix onnxruntime "pthread_setaffinity_np failed" error

encoder.py

@@ -22,7 +22,11 @@ class Encoder(object):
     ) -> None:
         self.input_name = 'input_0'
         self.output_name = 'output_0'
-        self.session = onnxruntime.InferenceSession(model_path, None)
+
+        opts = onnxruntime.SessionOptions()
+        opts.intra_op_num_threads = 8
+        self.session = onnxruntime.InferenceSession(
+            model_path, sess_options=opts)
 
     def predict(self, image: PIL.Image.Image) -> np.ndarray:
         x = to_tensor(image)

generator.py

@@ -18,7 +18,10 @@ class Generator(object):
     ) -> None:
         self.input_name = 'input_0'
         self.output_name = 'output_0'
-        self.session = onnxruntime.InferenceSession(model_path, None)
+        opts = onnxruntime.SessionOptions()
+        opts.intra_op_num_threads = 8
+        self.session = onnxruntime.InferenceSession(
+            model_path, sess_options=opts)
 
     def predict(self, x: np.ndarray) -> PIL.Image.Image:
         x = np.expand_dims(x, 0)