I run Transformer model in DGL and came across some error

Questions
#1

I have modified the file ‘translation_train.py’ as below according to my environment:

image
image1382×830 74.4 KB

And I got error as below:
image
image2782×1450 832 KB

This error may be caused by this function in models.py line 82:
image
image1360×392 55 KB

I have no idea to fix it.

#2

Hi, have you installed dgl with cuda support?

For example, if your cuda version is 10.0, you can install dgl with

conda install -c dglteam dgl-cuda10.0

or

pip install dgl-cu100