30 lines
901 B
Diff
30 lines
901 B
Diff
From bdf1875bcd4e34766d45a6514ec9fed9d2cf1a9a Mon Sep 17 00:00:00 2001
|
|
From: j1nx <p.steenbergen@j1nx.nl>
|
|
Date: Wed, 8 Jun 2022 11:44:11 +0200
|
|
Subject: [PATCH 1/1] Try to use 4 threads Tensorflow v2.3.0+ supports
|
|
multithreading.
|
|
|
|
---
|
|
precise_lite_runner/runner.py | 5 +++++
|
|
1 file changed, 5 insertions(+)
|
|
|
|
diff --git a/precise_lite_runner/runner.py b/precise_lite_runner/runner.py
|
|
index 7e56a30..0aadf73 100644
|
|
--- a/precise_lite_runner/runner.py
|
|
+++ b/precise_lite_runner/runner.py
|
|
@@ -16,6 +16,11 @@ class TFLiteRunner:
|
|
def __init__(self, model_name: str):
|
|
# Setup tflite environment
|
|
self.interpreter = tflite.Interpreter(model_path=model_name)
|
|
+ try:
|
|
+ self.interpreter.set_num_threads(4)
|
|
+ except:
|
|
+ pass
|
|
+
|
|
self.interpreter.allocate_tensors()
|
|
|
|
self.input_details = self.interpreter.get_input_details()
|
|
--
|
|
2.34.1
|
|
|