cjones/immich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tpe

Browse Source
This commit is contained in:
yoni13
2025-01-17 19:56:23 +08:00
parent bc48b67379
commit f067212491
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
machine-learning/app/sessions/rknn.py
+1 -1
View File
@@ -32,7 +32,7 @@ class RknnSession:
self.tpe = settings.rknn_facial_detection_threads self.tpe = settings.rknn_facial_detection_threads
log.info(f"Loading RKNN model from {self.model_path} with {self.tpe} 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.") log.info(f"Loaded RKNN model from {self.model_path} with {self.tpe} threads.")
def __del__(self) -> None: def __del__(self) -> None: