How to train on GPUs? - Githubissues

google-research / uda

Unsupervised Data Augmentation (UDA)

https://arxiv.org/abs/1904.12848

Apache License 2.0

2.17k stars 313 forks source link

Open vantony1 opened 3 years ago

vantony1 commented 3 years ago

Hello,

I am trying to run the training on GPU on my local machine but UDA won't recognize it. I am not using the TPU flag.

Is there any way to specify which GPU UDA should use?

Best