Running on CPU - Githubissues

AIPHES / DiscoScore

DiscoScore: Evaluating Text Generation with BERT and Discourse Coherence

32 stars 6 forks source link

Running on CPU #7

Closed UntotaufUrlaub closed 11 months ago

UntotaufUrlaub commented 11 months ago

When I set the device to "cpu" I get an error stating that not everything is on the same device. Can anybody confirm this error?

I had a brief look at the code and saw that "cuda:0" is the default at many places. Is the device maybe not passed at some point, falling back to the default?

andyweizhao commented 11 months ago

Hi, I just added the support for CPU. Please have a look at the latest commitment.