Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs
Apache License 2.0
695 stars 88 forks source link

Debug Assertion Failed #73

Open DonSeger opened 2 years ago

DonSeger commented 2 years ago

I have installed thundergbm with cuda 11 support. But when I try to run the thundergbm classifier inside a nested loop in order to do a hyperparameter search, I get the following error:

Debug Asertion Failed

Python Version: 3.7.8 Microsoft Visual Studio 19: v.16.11.8 Cuda: 11.5 cmake: 3.22.1

Below is the code that I run in a Python Jupyter Notebook:

for n in range(100,700,100):           
    for d in range(3,16,1):             
        for c in [0.4,0.5,0.6,0.7,0.8]:   

            clf = TGBMClassifier(depth=d, n_trees = 1, n_parallel_trees=n, bagging=1, column_sampling_rate=c, objective = 
                                             "binary:logistic")
            clf.fit(X_train, y_train)

What is the reason?