unexpected error, argument #3 'index' in call to _th_index_select

SamLynnEvans / Transformer

Transformer seq2seq model, program that can build a language translator from parallel corpus

Apache License 2.0

1.35k stars 350 forks source link

unexpected error, argument #3 'index' in call to _th_index_select #24

Open javismiles opened 4 years ago

javismiles commented 4 years ago

Im getting this error when following the readme instructions: RuntimeError: Expected object of device type cuda but got device type cpu for argument #3 'index' in call to _th_index_select

any tips?

(im running it in colab with cuda GPU)