remove openmp error msg on PC

Tencent · Dec 3, 2018 · 5ced530 · 5ced530
1 parent 9586866
commit 5ced530
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/net.cpp b/src/net.cpp
@@ -1013,6 +1013,9 @@ int Extractor::extract(int blob_index, Mat& feat)
     if (blob_mats[blob_index].dims == 0)
     {
         int layer_index = net->blobs[blob_index].producer;
+        if (opt.num_threads == 0) {
+            opt.num_threads = 1;
+        }
         ret = net->forward_layer(layer_index, blob_mats, opt);
     }