Open caprone opened 1 year ago
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.
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