seems don't run on GPU??

Open caprone opened this issue 3 years ago • 1 comments

HI i'm try to train a base model, but seems does't works with GPU?..is very slow and no output from (verbose =True)... any idea? Thanks

Feb 14 '23 19:02 caprone

Hi Sorry for the late reply. It does run on the GPU when GPU is available in your system. Did you check if you have CUDA? I am going to update a version here you can monitor the training process so you can see if the GPU is being used or not. Thank you.

Oct 27 '23 11:10 sunchang0124