How to use it in GPU? - Githubissues

google-research / bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.

https://arxiv.org/abs/2004.04696

Apache License 2.0

697 stars 85 forks source link

Closed g-jing closed 3 years ago

g-jing commented 4 years ago

The default choice seems to be CPU. How can I use GPU?

tsellam commented 4 years ago

Hi Roderick,

BLEURT runs on whichever device Tensorflow uses by default. We've never had to specify the GPU--TF should do it for you.