Error in training

[Luguo-ry]

Hi! I'm trying to train the neural network, using the default values provided in the script run_GAIN_BERT.sh, but the training script is giving me an error after beginning. It seems to be an error with DGL The error is the following:

[Emir-Liu]

when using the new version packet,I meet the problem too.And,I change the code in GAIN.py.

# graph_big = dgl.batch_hetero(graphs)
graph_big = dgl.batch_hetero(graphs).to('cuda:0')
...
# entity_graph_big = dgl.batch(entity_graphs)
entity_graph_big = dgl.batch(entity_graphs).to('cuda:0')

[LawsonAbs]

升级dgl库，然后改写对应代码，将模型放到GPU上就ok了

[duLun008]

when using the new version packet,I meet the problem too.And,I change the code in GAIN.py.

# graph_big = dgl.batch_hetero(graphs)
graph_big = dgl.batch_hetero(graphs).to('cuda:0')
...
# entity_graph_big = dgl.batch(entity_graphs)
entity_graph_big = dgl.batch(entity_graphs).to('cuda:0')

useful, and dgl-cu102 (0.4.3) not suitable. I use 0.6.1 then it works