search

chenxiaowei-vincent / XGraphBoost

56 stars 15 forks source link

Not able to reproduce the results: Single process multi-GPU training is no longer supported #1

Open sachin-vish91 opened 3 years ago

sachin-vish91 commented 3 years ago

I like the idea of applying the D-MPNN features for the XGBoost. Getting the following error while trying to reproduce the below command from the GitHub

python train.py --protein delaney_ESOL --dataset_type regression --save_dir delaney_ESOL_r --epoch 200

Could you please help me with this I am running this on a CPU machine. I followed the instruction given in the GitHub for the installation of the package.

Error

chenxiaowei-vincent commented 3 years ago

hello，You can delete it. 'N_GPUS =-1'

At 2021-06-18 18:13:59, "Sachin Vishwakarma" @.***> wrote:

I like the idea of applying the D-MPNN features for the XGBoost. Getting the following error while trying to reproduce the below command from the GitHub

python train.py --protein delaney_ESOL --dataset_type regression --save_dir delaney_ESOL_r --epoch 200

Could you please help me with this I am running this on a CPU machine. I followed the instruction given in the GitHub for the installation of the package.

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.

sachin-vish91 commented 3 years ago

Thanks, it worked.