Skip to content

Commit

Permalink
bruh i forgor to accept the user defined core
Browse files Browse the repository at this point in the history
  • Loading branch information
james20902 committed Feb 2, 2024
1 parent 2db8b1b commit 8280977
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/main/native/cpp/yolo_common.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -83,7 +83,7 @@ YoloModel::YoloModel(std::string modelPath, int num_classes_, ModelVersion type_
}

// hard coded to let npu decide where the model runs
this->changeCoreMask(-1);
this->changeCoreMask(coreNumber);

rknn_sdk_version version;
ret = rknn_query(ctx, RKNN_QUERY_SDK_VERSION, &version, sizeof(rknn_sdk_version));
Expand Down

0 comments on commit 8280977

Please sign in to comment.