Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix omp

Browse Source
This commit is contained in:
tqchen 2014-08-18 08:57:26 -07:00
parent 0d9a8c042c
commit e912dd3364
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/gbm/gbtree-inl.hpp
View File

@ -287,8 +287,7 @@ class GBTree : public IGradBooster<FMatrix> {
updater_initialized = 0; updater_initialized = 0;
} }
if (!strcmp(name, "nthread")) { if (!strcmp(name, "nthread")) {
omp_set_num_threads(nthread); omp_set_num_threads(nthread = atoi(val));
nthread = atoi(val);
} }
if (!strcmp(name, "num_parallel_tree")) { if (!strcmp(name, "num_parallel_tree")) {
num_parallel_tree = atoi(val); num_parallel_tree = atoi(val);