• shiyu1994's avatar
    Add quantized training (CPU part) (#5800) · 17ecfab3
    shiyu1994 authored
    * add quantized training (first stage)
    
    * add histogram construction functions for integer gradients
    
    * add stochastic rounding
    
    * update docs
    
    * fix compilation errors by adding template instantiations
    
    * update files for compilation
    
    * fix compilation of gpu version
    
    * initialize gradient discretizer before share states
    
    * add a test case for quantized training
    
    * add quantized training for data distributed training
    
    * Delete origin.pred
    
    * Delete ifelse.pred
    
    * Delete LightGBM_model.txt
    
    * remove useless changes
    
    * fix lint error
    
    * remove debug loggings
    
    * fix mismatch of vector and allocator types
    
    * remove changes in main.cpp
    
    * fix bugs with uninitialized gradient discretizer
    
    * initialize ordered gradients in gradient discretizer
    
    * disable quantized training with gpu and cuda
    
    fix msvc compilation errors and warnings
    
    * fix bug in data parallel tree learner
    
    * make quantized training test deterministic
    
    * make quantized training in test case more accurate
    
    * refactor test_quantized_training
    
    * fix leaf splits initialization with quantized training
    
    * check distributed quantized training result
    17ecfab3
multi_val_sparse_bin.hpp 18.1 KB