处理one-hop节点的代码的一个疑问 about paperrobot HOT 4 CLOSED

Hi @WongIhsin, Thank you very much for your interest in our research and the suggestion. I have fixed this bug for the following commit. If you have further questions, you can also contact me using the email on my Github page. Hope you have a great day!

from paperrobot.

抱歉我的英文不是很好，所以还是用中文来回复了。
我觉得get_subgraph()函数中head和tail可以先head = triple[0].item()，这样可以避免included_nodes中加入tensor类型的nodes，使得adj_ingraph矩阵有多余的0行；
另外，在batch_size为200时候，included_nodes的范围大约会在10000左右，以至于adj_ingraph这个矩阵会很大，计算attention时候很容易显存溢出，并且因为shifted_pos还会进行截取，所以这里会造成很大的资源浪费。
我还阅读了https://github.com/Diego999/pyGAT的代码，发现pyGAT中的adj矩阵比较小，并不会有显存溢出的问题，并且pyGAT中的attention在更新权重的时候会都使用到。
所以我想是不是可以将shifted_pos的截取这一步移动到leakyrelu之前，可以很大程度的减少显存消耗

from paperrobot.

@WongIhsin 对于第一个问题我已经fix了，对第二个问题，我觉得可以把to(device)放到shifted_pos, shifted_neg = convert_index([pos, neg], batch_nodes)后应该会解决一定问题吧。还有就是batchsize可以改小一点

from paperrobot.

好的，非常感谢😀

from paperrobot.