multitask-learning-transformers icon indicating copy to clipboard operation
multitask-learning-transformers copied to clipboard

using cpu not gpu

Open hxtkyne opened this issue 3 years ago • 2 comments

Thanks for you work. I have a problem: the training using cpu not gpu.

hxtkyne avatar Feb 17 '22 09:02 hxtkyne

@hxtkyne Could you please tell me what exact error are you getting? Also, note that training using CPU will be considerably slower.

shahrukhx01 avatar Feb 17 '22 09:02 shahrukhx01

@hxtkyne Could you please tell me what exact error are you getting? Also, note that training using CPU will be considerably slower.

I doesn't not get any error. I can use your code. But the training doesn't use gpu, in fact, I have 2 gpus.

hxtkyne avatar Feb 17 '22 09:02 hxtkyne