search

graykode / nlp-tutorial

Natural Language Processing Tutorial for Deep Learning Researchers
https://www.reddit.com/r/MachineLearning/comments/amfinl/project_nlptutoral_repository_who_is_studying/
MIT License
14.03k stars 3.9k forks source link

Transformer/Transformer(Greedy_decoder)-Torch.py on gpu #35

Open kangkang61 opened 5 years ago

kangkang61 commented 5 years ago

Hello, I want to put the Transformer (Greedy_decoder)-Torch.py code on the gpu, using model=model.to(device), input_data also to (device), but the error still appears "Expected object of backend CUDA but backend CPU for argument #2 'mat2”

zhangbo2008 commented 4 years ago

i fix the code in my clones https://github.com/zhangbo2008/nlp-tutorial/blob/master/1gpu.py use:pytorch 1.4 can run

zhangbo2008 commented 4 years ago

i am learning transformer too , i still wonder why we use target_sentence in traning?