I am getting this error message
  File "main.py", line 185, in <module>
    input = input.to(device)
RuntimeError: CUDA error: device-side assert triggered

I try commenting out these lines of code;

#if device == torch.device(“cuda”):
#input = input.to(device)
output = skipgram_model(input)

But then I get this error message;
RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'index' in call to _th_index_select
How do I fix this?