Add warnings for large labels when using GPU histogram algorithms (#2834)

2017-10-26 17:31:10 +13:00
parent 13e7a2cff0
commit d9d5293cdb
5 changed files with 51 additions and 23 deletions
--- a/src/tree/updater_gpu_hist.cu
+++ b/src/tree/updater_gpu_hist.cu
@@ -537,6 +537,9 @@ class GPUHistMaker : public TreeUpdater {
      device_gpair[d_idx].copy(gpair.begin() + device_row_segments[d_idx],
                               gpair.begin() + device_row_segments[d_idx + 1]);

+      // Check gradients are within acceptable size range
+      CheckGradientMax(device_gpair[d_idx]);
+
      subsample_gpair(&device_gpair[d_idx], param.subsample,
                      device_row_segments[d_idx]);