Skip to content

Conversation

zergzzlun
Copy link

Hi there.

Using CUDA version of local spatial encoding in the 'gather' branch (#17) will get the following error:

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking arugment for argument index in method wrapper_gather)

This can be fixed by moving indexes (expanded_idx) to the same device of features.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant