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

minor fix: log InitDataOnce() only when it is actually called (#4206)

Browse Source
This commit is contained in:
Rong Ou 2019-03-07 13:53:09 -08:00 committed by Rory Mitchell
parent ac3d03089b
commit 0944360416
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tree/updater_gpu_hist.cu
View File

@ -1029,11 +1029,11 @@ class GPUHistMakerSpecialised{
}
void InitData(HostDeviceVector<GradientPair>* gpair, DMatrix* dmat) {
monitor_.Start("InitDataOnce", dist_.Devices());
if (!initialised_) {
monitor_.Start("InitDataOnce", dist_.Devices());
this->InitDataOnce(dmat);
monitor_.Stop("InitDataOnce", dist_.Devices());
}
monitor_.Stop("InitDataOnce", dist_.Devices());
column_sampler_.Init(info_->num_col_, param_.colsample_bynode,
param_.colsample_bylevel, param_.colsample_bytree);