yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-11-20 12:03:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix rec v3 mkldnn irpass in cpp infer (#6154)

Browse Source 
* fix rec v3 mkldnn irpass

* fix rec v3 mkldnn irpass
This commit is contained in:
andyjpaddle 2022-05-06 21:57:41 +08:00 committed by GitHub
parent 12aa5e80fa
commit 369889b6e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deploy/cpp_infer/src/ocr_rec.cpp
View File

@ -166,6 +166,10 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
} }
// get pass_builder object
auto pass_builder = config.pass_builder();
// delete "matmul_transpose_reshape_fuse_pass"
pass_builder->DeletePass("matmul_transpose_reshape_fuse_pass");
config.SwitchUseFeedFetchOps(false); config.SwitchUseFeedFetchOps(false);
// true for multiple input // true for multiple input
config.SwitchSpecifyInputNames(true); config.SwitchSpecifyInputNames(true);