From f0672124917415290f8cf7a8dd3b226c8f17c870 Mon Sep 17 00:00:00 2001 From: yoni13 Date: Fri, 17 Jan 2025 19:56:23 +0800 Subject: [PATCH] tpe --- machine-learning/app/sessions/rknn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/machine-learning/app/sessions/rknn.py b/machine-learning/app/sessions/rknn.py index 585129c0ea..fd79884be6 100644 --- a/machine-learning/app/sessions/rknn.py +++ b/machine-learning/app/sessions/rknn.py @@ -32,7 +32,7 @@ class RknnSession: self.tpe = settings.rknn_facial_detection_threads log.info(f"Loading RKNN model from {self.model_path} with {self.tpe} threads.") - self.rknnpool = RknnPoolExecutor(rknnModel=self.model_path.as_posix(), TPEs=self.tpe, func=runInfrence) + self.rknnpool = RknnPoolExecutor(rknnModel=self.model_path.as_posix(), tpes=self.tpe, func=runInfrence) log.info(f"Loaded RKNN model from {self.model_path} with {self.tpe} threads.") def __del__(self) -> None: