Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
fd66c409
"...composable_kernel_onnx.git" did not exist on "827301d95af93d581ddac8d2734ec759ea215c6c"
Commit
fd66c409
authored
Jul 13, 2020
by
littletomatodonkey
Browse files
rm mkl
parent
51a8d233
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
deploy/cpp_infer/src/ocr_det.cpp
deploy/cpp_infer/src/ocr_det.cpp
+1
-1
deploy/cpp_infer/src/ocr_rec.cpp
deploy/cpp_infer/src/ocr_rec.cpp
+1
-1
No files found.
deploy/cpp_infer/src/ocr_det.cpp
View file @
fd66c409
...
@@ -25,7 +25,7 @@ void DBDetector::LoadModel(const std::string &model_dir) {
...
@@ -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_
);
}
}
...
...
deploy/cpp_infer/src/ocr_rec.cpp
View file @
fd66c409
...
@@ -128,7 +128,7 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
...
@@ -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_
);
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment