BuboQA
BuboQA copied to clipboard
Allow `top_retrieval.py` to handle `--no_cuda`
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.