From 1795bb1d08dab5c7e6c30e908b2c36e84f8c9426 Mon Sep 17 00:00:00 2001 From: andyjpaddle Date: Fri, 6 May 2022 11:58:36 +0000 Subject: [PATCH] fix rec v3 mkldnn irpass --- deploy/cpp_infer/src/ocr_rec.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/deploy/cpp_infer/src/ocr_rec.cpp b/deploy/cpp_infer/src/ocr_rec.cpp index 298b16bf..f69f37b8 100644 --- a/deploy/cpp_infer/src/ocr_rec.cpp +++ b/deploy/cpp_infer/src/ocr_rec.cpp @@ -166,9 +166,9 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) { config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); } - // 得到 pass_builder 对象 + // get pass_builder object auto pass_builder = config.pass_builder(); - // 在 IR 优化阶段，去除 fc_fuse_pass + // delete "matmul_transpose_reshape_fuse_pass" pass_builder->DeletePass("matmul_transpose_reshape_fuse_pass"); config.SwitchUseFeedFetchOps(false); // true for multiple input -- GitLab