stevezheng23 / xlnet_extension_tf

XLNet Extension in TensorFlow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Using GPU

mayinghan opened this issue · comments

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.

By default, GPU should be used if there's any