Initial CUDA work

microsoft · Jun 11, 2020 · e1f8270 · e1f8270
1 parent 8b5e5e3
commit e1f8270
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/boosting/gbdt.cpp b/src/boosting/gbdt.cpp
@@ -91,6 +91,8 @@ void GBDT::Init(const Config* config, const Dataset* train_data, const Objective
     num_tree_per_iteration_ = objective_function_->NumModelPerIteration();
   }
 
+  is_constant_hessian_ = GetIsConstHessian(objective_function);
+
   // push training metrics
   training_metrics_.clear();
   for (const auto& metric : training_metrics) {