diff --git a/machine-learning/rknn/rknnpool.py b/machine-learning/rknn/rknnpool.py index b248f28ccb..e8a3a8bcea 100644 --- a/machine-learning/rknn/rknnpool.py +++ b/machine-learning/rknn/rknnpool.py @@ -55,7 +55,7 @@ def init_rknn(rknnModel, id) -> RKNNLite: if ret != 0: raise RuntimeError("Init runtime environment failed") - print(rknnModel, "\t\tdone") + return rknn_lite