forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

rm mkl

This commit is contained in:
littletomatodonkey 2020-07-13 14:09:04 +00:00
parent 51a8d23392
commit fd66c409b0
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
deploy/cpp_infer/src/ocr_det.cpp
View File

@ -25,7 +25,7 @@ void DBDetector::LoadModel(const std::string &model_dir) {
config.EnableUseGpu(this->gpu_mem_, this->gpu_id_); config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
} else { } else {
config.DisableGpu(); config.DisableGpu();
config.EnableMKLDNN(); // 开启MKLDNN加速 // config.EnableMKLDNN(); // not sugesteed to use for now
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
} }

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

@ -128,7 +128,7 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
config.EnableUseGpu(this->gpu_mem_, this->gpu_id_); config.EnableUseGpu(this->gpu_mem_, this->gpu_id_);
} else { } else {
config.DisableGpu(); config.DisableGpu();
config.EnableMKLDNN(); // 开启MKLDNN加速 // config.EnableMKLDNN(); // not sugesteed to use for now
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
} }