BuboQA icon indicating copy to clipboard operation
BuboQA copied to clipboard

Allow `top_retrieval.py` to handle `--no_cuda`

Open amadanmath opened this issue 6 years ago • 0 comments

Currently, if either entity or relation top_retrieval.py are invoked with --no_cuda, it breaks with the following error:

RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 'index'

The reason is the use of CUDA storage regardless of whether or not CUDA has been enabled.

amadanmath avatar May 28 '19 10:05 amadanmath