search

shahrukhx01 / multitask-learning-transformers

A simple recipe for training and inferencing Transformer architecture for Multi-Task Learning on custom datasets. You can find two approaches for achieving this in this repo.
Apache License 2.0
87 stars 11 forks source link

using cpu not gpu #2

Open hxtkyne opened 2 years ago

hxtkyne commented 2 years ago

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

shahrukhx01 commented 2 years ago

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

hxtkyne commented 2 years ago

@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.