Thanks for sharing! I just made a test-run with the current torch docker container.
However I get an error in validation.py", line 78, in _topk_retrieval where
retrieval = torch.gather(candidates, 1, yi)RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
I may be running a different torch version. Can you add your python version / requirements.txt ?
Thanks for sharing! I just made a test-run with the current torch docker container. However I get an error in
validation.py", line 78, in _topk_retrieval
whereretrieval = torch.gather(candidates, 1, yi)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
I may be running a different torch version. Can you add your python version / requirements.txt ?
Best