Using GPU
mayinghan opened this issue · comments
Yinghan Ma commented
Hi,
Im trying to run NER task, but I think the model training is running on CPU instead of GPU. Is there any way to train on GPU, I saw the only choice in the flag is TPU on or off.
Xiaoming commented
By default, GPU should be used if there's any