About the module train
Opened this issue · 0 comments
xxxmmm00 commented
Hello！Do you train your model on GPU? During training, I found that the code only runs on the CPU. Now I want to put the code on the GPU. Is there any way?
Opened this issue · 0 comments
Hello！Do you train your model on GPU? During training, I found that the code only runs on the CPU. Now I want to put the code on the GPU. Is there any way?