更新cls的推理修复

2026-01-08 17:42:25 +08:00
parent d5457e02ca
commit 2a185affc3
1 changed files with 1 additions and 1 deletions
--- 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