From 2a185affc377d47c224c23b11ae9d3e6003e9f00 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=90=89=E7=92=83=E6=9C=88=E5=85=89?= <15630071+llyg777@user.noreply.gitee.com> Date: Thu, 8 Jan 2026 17:42:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0cls=E7=9A=84=E6=8E=A8?= =?UTF-8?q?=E7=90=86=E4=BF=AE=E5=A4=8D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- cls_main/main_cls.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cls_main/main_cls.py b/cls_main/main_cls.py index f5c66f6..f184051 100644 --- a/cls_main/main_cls.py +++ b/cls_main/main_cls.py @@ -46,7 +46,7 @@ def init_rknn_model(model_path): ret = rknn_lite.load_rknn(model_path) if ret != 0: raise RuntimeError(f'Load model failed: {ret}') - ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0) + ret = rknn_lite.init_runtime() if ret != 0: raise RuntimeError(f'Init runtime failed: {ret}') _global_rknn_instance = rknn_lite