Skip to content

Conversation

@Abhijit-2592
Copy link
Contributor

When training on GPU, the embedding layer throws the following error:

RuntimeError: Expected object of device type cuda but got device type cpu for argument # 3 'index' in call to _th_index_select

This is fixed by casting the CPU Tensors to CUDA Tensors. Also added the line which says to move the created conll files inside separate sub-directories inside data directory before parsing it with spaCy

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants