search

shmsw25 / bart-closed-book-qa

A BART version of an open-domain QA model in a closed-book setup
119 stars 14 forks source link

Adds a check for GPU to allow CPU runtime #2

Closed idocal closed 4 years ago

idocal commented 4 years ago

It's best practice to check for cuda availability in order to be able to run the code on machines without GPU support.